[JAX] Replace uses of jax.experimental.pjit.with_sharding_constraint … #9959
build.yml
on: push
Matrix: Test import standalone
Cancel Previous Runs
2s
Test pre-commit hooks
30s
Check commit count
8s
Matrix: Run Tests
Annotations
5 warnings