7 projects
fleqx
Normalizing flows for JAX, with a distreqx-native API.
paramrf
Parametric radio frequency modeling
parax
Parametric modeling in JAX
refrax
Chainable optics for JAX PyTrees
eqxpress
Composable expressions in JAX + Equinox
inferix
A unified interface for probabilistic inference in JAX + Equinox
eqx-learn
Classical machine learning using JAX and Equinox
