JAX-Triton

https://github.com/jax-ml/jax-triton

JAX-Triton enables writing custom GPU kernels using Triton that can be embedded inside JAX programs...

Tokens:40,438
Snippets:404
Trust Score:8.4
License:Apache-2.0
Update:2 weeks ago
Tokens:
Raw