GitHub repository: m-a-n-i-f-e-s-t/jax-triton
Primary language: Python
Description: jax-triton contains integrations between JAX and OpenAI Triton