Unittests

Update pathways.experimental.reshard so that PyTrees with arrays that… #324

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Unittests (Python 3.10, JAX 0.5.3)	30s
Unittests (Python 3.11, JAX 0.5.3)	28s
Unittests (Python 3.12, JAX 0.5.3)	31s
Unittests (Python 3.12, JAX 0.6.1)	29s
Unittests (Python 3.11, JAX 0.6.0)	29s
Unittests (Python 3.10, JAX 0.6.0)	29s
Unittests (Python 3.10, JAX 0.6.1)	26s
Unittests (Python 3.12, JAX 0.7.0)	39s
Unittests (Python 3.11, JAX 0.6.1)	40s
Unittests (Python 3.11, JAX 0.6.2)	30s
Unittests (Python 3.10, JAX 0.6.2)	31s
Unittests (Python 3.11, JAX 0.7.0)	30s
Unittests (Python 3.12, JAX 0.6.2)	32s
Unittests (Python 3.12, JAX 0.6.0)	29s
	7m 13s