GitHub repository: tonywu95/jax-triton
Description: jax-triton contains integrations between JAX and OpenAI Triton