Skip to content

Add guard at lowest JAX version that still supports triton kernel calling#2741

Open
tdophung wants to merge 2 commits intoNVIDIA:mainfrom
tdophung:triton_jax_bwd_compat
Open

Add guard at lowest JAX version that still supports triton kernel calling#2741
tdophung wants to merge 2 commits intoNVIDIA:mainfrom
tdophung:triton_jax_bwd_compat

Commits

Commits on Mar 6, 2026