software generax A JAX and Equinox library for normalizing flows with practical support for tabular vectors and images. linsdex A high-performance JAX library for linear stochastic differential equations, state space models, and Gaussian inference. local_coordinates A JAX framework for differential geometry computations on Riemannian manifolds.