search

jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
354 stars 40 forks source link

Introduce GPU testing to JAX Triton #298

Closed MichaelHudgins closed 1 month ago

MichaelHudgins commented 2 months ago

Introduce GPU testing presubmits into JAX triton