Last released Jul 12, 2024
A simple worker library in Python
Last released Jul 7, 2024
Converts torch models into PyTrees for Equinox
Last released Jun 21, 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