Last released Jun 16, 2024
Converts torch models into PyTrees for Equinox
Last released Jun 13, 2024
JAX models for deep learning
Last released May 18, 2024
Functions for JAX
Last released May 17, 2024
A dataloader, but for JAX/Numpy
Last released Mar 5, 2024
That's right, I'm Kira ✍️
Last released Oct 8, 2023
Supported by