jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
354
stars
40
forks
source link
Integrate Triton up to [68aa962e67baa191cec5aac173255abdba80db1a](https://github.com/openai/triton/commits/68aa962e67baa191cec5aac173255abdba80db1a) #313
Integrate Triton up to 68aa962e67baa191cec5aac173255abdba80db1a