Home / JAX / Neural Network Libraries > Elegy Neural Network Libraries > Elegy A High Level API for Deep Learning in JAX. Supports Flax, Haiku, and Optax. Package 476 stars GitHub Back to JAX