Development fork of `jax-triton`: integrations between JAX and OpenAI Triton - View it on GitHub