Skip to content

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

Open
tdophung wants to merge 9 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 9 commits intoNVIDIA:mainfrom
tdophung:triton_jax_bwd_compat

Commits

Commits on Mar 6, 2026

Commits on Mar 9, 2026

Commits on Mar 10, 2026