search

jax-ml / jax-triton

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

WIP - testing out JAX fix #270

Closed copybara-service[bot] closed 9 months ago

copybara-service[bot] commented 9 months ago

WIP - testing out JAX fix