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

Star on GitHub
jax-ml

jax-triton

jax-ml/jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

444stars
View on GitHub
Forks
57
Open issues
30
Watchers
444
Size
1.2 MB
PythonApache License 2.0
Created: Jul 28, 2022
Updated: Apr 11, 2026
Last push: Mar 26, 2026