Last released Aug 13, 2026
JAX + OpenAI Triton integration
Last released Mar 20, 2026
A gradient processing and optimization library in JAX.
Supported by