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

Star on GitHub
jax-ml

jax-triton

jax-ml/jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

472stars
View on GitHub
Forks
60
Open issues
12
Watchers
472
Size
1.2 MB
PythonApache License 2.0
Created: Jul 28, 2022
Updated: Oct 6, 2026
Last push: Oct 9, 2026