Last released Jun 10, 2024
Easily convert iterables over large datasets into random access.
Last released Nov 10, 2023
automatic mixed precision in JAX
Last released Sep 8, 2023
black box tuning of optimizers
Last released Mar 17, 2023
pytorch-like ML with Jax
Supported by
