Last released Sep 14, 2025
A neural network library for JAX that combines functional init/apply style with a flexible model definition.
Supported by
