jax-triton

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

Downloads last day: 688
Downloads last week: 6,650
Downloads last month: 28,117