Skip to content

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

Merged
tdophung merged 9 commits intoNVIDIA:mainfrom
tdophung:triton_jax_bwd_compat
Mar 11, 2026
Merged

Add guard at lowest JAX version that still supports triton kernel calling#2741
tdophung merged 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