Last released Feb 28, 2026
Flash Attention port for JAX
Last released Sep 10, 2025
A jax based nn library
Last released Feb 24, 2024
A simple jax implementation of CLIP using jaxtorch.
Supported by
