Containerized Training Differences Tests Jax vs PyTorch

Migrate JAX workloads from pmap to jit #175

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
build_and_push_docker_image	6m 1s
traindiffs_tests	0s
	6m 1s