jax-triton

PyPI page
Summary: JAX + OpenAI Triton integration
Latest version: 0.4.1
Required dependencies: absl-py | jax | triton
Optional dependencies: pytest

Downloads last day: 736
Downloads last week: 6,763
Downloads last month: 36,428
7-day smoothing