Last released Jul 6, 2023
A simple deep learning library for JAX.
A JAX Neural Simulator
Supported by