Code

Add upstream pax 2-node Grok test#854

Code

Open

ashors1 wants to merge 21 commits intomainNVIDIA/JAX-Toolbox:mainfrom pax-grok-ciNVIDIA/JAX-Toolbox:pax-grok-ciCopy head branch name to clipboard

+174-174

Commits

Commits on May 24, 2024

• add upstream pax 2-node Grok test

  ashors1

  committed
  1bb2d16

  Copy full SHA for 1bb2d16

Commits on May 28, 2024

• Merge branch 'main' of github.com:NVIDIA/JAX-Toolbox into pax-grok-ci

  ashors1

  committed
  c9d1d5b

  Copy full SHA for c9d1d5b

Commits on May 29, 2024

• update pax tests to pass in ICI and DCN mesh shapes explicitly

  ashors1

  committed
  17a7eb4

  Copy full SHA for 17a7eb4

Commits on May 30, 2024

• fixes

  ashors1

  committed
  374a6b0

  Copy full SHA for 374a6b0

• fix

  ashors1

  committed
  2b87465

  Copy full SHA for 2b87465

• fix

  ashors1

  committed
  67dbc37

  Copy full SHA for 67dbc37

• fix

  ashors1

  committed
  86f043a

  Copy full SHA for 86f043a

• fix syntax

  ashors1

  committed
  0ac6db7

  Copy full SHA for 0ac6db7

• fix ICI and DCN mesh shapes

  ashors1

  committed
  e6c9aeb

  Copy full SHA for e6c9aeb

Commits on May 31, 2024

• more fixes

  ashors1

  committed
  f226c03

  Copy full SHA for f226c03

• fix max_steps

  ashors1

  committed
  e5bd851

  Copy full SHA for e5bd851

• fix

  ashors1

  committed
  817e351

  Copy full SHA for 817e351

• fix command line args

  ashors1

  committed
  e98cc98

  Copy full SHA for e98cc98

• cleanup, bug fixes

  ashors1

  committed
  c53fab3

  Copy full SHA for c53fab3

• attempt tp fix ici and dcn mesh shape formatting

  ashors1

  committed
  ad61b48

  Copy full SHA for ad61b48

Commits on Jun 1, 2024

• fix mesh shapes

  ashors1

  committed
  a64bfad

  Copy full SHA for a64bfad

Commits on Jun 3, 2024

• ici/dcn syntax fix

  ashors1

  committed
  5101dbb

  Copy full SHA for 5101dbb

• fix mesh shapes

  ashors1

  committed
  9e3b1aa

  Copy full SHA for 9e3b1aa

• fix

  ashors1

  committed
  99e8ca6

  Copy full SHA for 99e8ca6

Commits on Jun 4, 2024

• bug fixes

  ashors1

  committed
  d8ea091

  Copy full SHA for d8ea091

Commits on Jun 7, 2024

• reduce num_layers for grok

  ashors1

  committed
  3b8a66c

  Copy full SHA for 3b8a66c