Add MNIST example with SPMD for JAX #2041

Job	Run time
Publish Image (dataset-initializer-v2, cmd/initializer_v2/dataset/Dockerfile, linux/amd64,linux/a... / Publish Image	2m 21s
Publish Image (kubectl-delivery, build/images/kubectl-delivery/Dockerfile, linux/amd64,linux/arm6... / Publish Image	2m 7s
Publish Image (model-initializer-v2, cmd/initializer_v2/model/Dockerfile, linux/amd64,linux/arm64... / Publish Image	3m 40s
Publish Image (storage-initializer, sdk/python/kubeflow/storage_initializer/Dockerfile, sdk/pytho... / Publish Image	10m 19s
Publish Image (trainer-huggingface, sdk/python/kubeflow/trainer/Dockerfile, sdk/python/kubeflow/t... / Publish Image	11m 11s
Publish Image (training-operator, build/images/training-operator/Dockerfile, linux/amd64,linux/ar... / Publish Image	29m 25s
Publish Image (training-operator-v2, cmd/training-operator.v2alpha1/Dockerfile, linux/amd64,linux... / Publish Image	25m 20s
	1h 24m 23s

Provide feedback