Last released Oct 21, 2024
Parameterizations and parameter constraints for JAX PyTrees.
Last released Oct 17, 2024
Easy to use distributions, bijections and normalizing flows in JAX.
Supported by