[JAX API] Updating `TransferToMemoryKind` and `jax.experimental.pallas.triton` - #1339

Closed

Steboss wants to merge 6 commits into

Commits on Feb 2, 2026

Fix memorykind error
Steboss
committed
c186135
update triton to latest changes
Steboss
committed
fe7eda8
temporary fix for envy
Steboss
committed
0462107
add a check on jax version to solve the version error
Steboss
committed
f1ac007
modify pr
Steboss
committed
18aebfd
use another strategy to get jax version'
Steboss
committed
2728a5f