audax. A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.