search

jax-ml / jax-triton

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

[jax_triton] Add parameter allowing user to compile for specific compute capability. #268

Closed copybara-service[bot] closed 9 months ago

copybara-service[bot] commented 9 months ago

[jax_triton] Add parameter allowing user to compile for specific compute capability.