[JAX] Replace uses of jax.experimental.pjit.with_sharding_constraint … #9959

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Test pre-commit hooks	30s
Cancel Previous Runs	2s
Check commit count	8s
Test import standalone (3.9)	35s
Test import standalone (3.10)	34s
Test import standalone (3.11)	36s
Run Tests (3.9, doctest)	4m 5s
Run Tests (3.9, pytest)	5m 0s
Run Tests (3.9, pytype)	6m 16s
Run Tests (3.9, mypy)	55s
Run Tests (3.10, doctest)	4m 12s
Run Tests (3.10, pytest)	4m 52s
Run Tests (3.10, mypy)	1m 6s
Run Tests (3.11, doctest)	4m 4s
Run Tests (3.11, pytest)	4m 58s
	37m 53s