will-thompson-k / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

Geek Repo

Github PK Tool