Containerized Training Differences Tests Jax vs PyTorch

Migrate JAX workloads from pmap to jit #173

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
build_and_push_docker_image	16m 2s
traindiffs_tests	0s
	16m 2s