Skip to content

fix: pjit_p for jax v0.7+#70

Merged
patrick-kidger merged 3 commits intopatrick-kidger:mainfrom
nstarman:fix-pjit-jax0.7+
Aug 24, 2025
Merged

fix: pjit_p for jax v0.7+#70
patrick-kidger merged 3 commits intopatrick-kidger:mainfrom
nstarman:fix-pjit-jax0.7+

Commits

Commits on Aug 22, 2025