JAX

RHMF in JAX. PCA-like matrix factorisation that treats bad data and outliers.

Composable latent-field models for IFU spectroscopy.