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