Last released May 24, 2024
Flash Attention Jax
Last released Feb 24, 2024
A simple jax implementation of CLIP using jaxtorch.
Last released Oct 4, 2022
A jax based nn library
Supported by
