JAX

Robust heteroscedastic matrix factorisation in JAX.

Composable statistical models for IFU spectroscopy, built on JAX.