5 projects
sbgm
Score-based Diffusion models in JAX.
transformer-flows
Implementation of Transformer Flows (Apple ML) in JAX and Equinox.
rectified-flows
Implementation of Rectified Flows in JAX and Equinox.
transformer_flow
Implementation of Apple ML's TARFlow in JAX.
sbiax
Fast, parallel and lightweight simulation-based inference in JAX.
