⭐ Star AlbumentationsX on GitHub — 448+ stars and counting!

Star on GitHub
jax-ml

jax-triton

jax-ml/jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

460stars
View on GitHub
Forks
57
Open issues
35
Watchers
460
Size
1.2 MB
PythonApache License 2.0
Created: Jul 28, 2022
Updated: May 22, 2026
Last push: Apr 23, 2026