Last released Apr 9, 2024
Functionalities such as a layers for building neural networks in Jax.
Framework to configure and run machine learning experiments on top of Jax.
Supported by