Similar Items: SBGM: Score-Based Generative Models in JAX.
- scarlet2: Astronomical scene modeling in JAX
- JAXtronomy: A JAX port of lenstronomy
- JAX-bandflux: differentiable supernovae SALT modelling for cosmological analysis on GPUs
- Wasserstein Convergence Guarantees for a General Class of Score-Based Generative Models
- J-UBIK: The JAX-accelerated Universal Bayesian Imaging Kit
- Score-Based Diffusion Models in Function Space