Code

[WIP] Testing CI in JAX example#2385

Code

Open

saileshd1402 wants to merge 13 commits intokubeflow:masterkubeflow/training-operator:masterfrom saileshd1402:jax-examplesaileshd1402/training-operator:jax-exampleCopy head branch name to clipboard

+494-9

Commits

Commits on Dec 25, 2024

• Add MNIST example with SPMD for JAX

  Show description for 619aa20
  sandipanpanda

  committed
  619aa20

  Copy full SHA for 619aa20

• Update CONTRIBUTING.md

  Show description for ca78118
  sandipanpanda

  committed
  ca78118

  Copy full SHA for ca78118

• Add JAXJob output

  Show description for 7f523e4
  sandipanpanda

  committed
  7f523e4

  Copy full SHA for 7f523e4

• Update JAXJob CI images

  Show description for 1ed9b7d
  sandipanpanda

  committed
  1ed9b7d

  Copy full SHA for 1ed9b7d

Commits on Dec 28, 2024

• Adjust jaxjob spmd example batch size

  Show description for 904dbd6
  sandipanpanda

  committed
  904dbd6

  Copy full SHA for 904dbd6

Commits on Jan 9, 2025

• Add JAX Example Docker Image Build in CI

  Show description for ef997a1
  saileshd1402

  committed
  ef997a1

  Copy full SHA for ef997a1

• Fix script name typo

  Show description for 9a88072
  saileshd1402

  committed
  9a88072

  Copy full SHA for 9a88072

• Update script permissions

  Show description for a2ca3a3
  saileshd1402

  committed
  a2ca3a3

  Copy full SHA for a2ca3a3

• Add KIND_CLUSTER env var

  Show description for 15e6205
  saileshd1402

  committed
  15e6205

  Copy full SHA for 15e6205

• Increase timeouts

  Show description for 07688ec
  saileshd1402

  committed
  07688ec

  Copy full SHA for 07688ec

Commits on Jan 10, 2025

• Test higher resources

  saileshd1402

  committed
  145d7fc

  Copy full SHA for 145d7fc

• Increase Timeout

  saileshd1402

  committed
  d2b7314

  Copy full SHA for d2b7314

• remove resource reqs

  Show description for 9edbd8f
  saileshd1402

  committed
  9edbd8f

  Copy full SHA for 9edbd8f