Code

Add GKE A3 Ultra support#940

Code

Open

samos123 wants to merge 6 commits intoapple:mainapple/axlearn:mainfrom samos123:a3u-supportsamos123/axlearn:a3u-supportCopy head branch name to clipboard

+334-140

Commits

Commits on Jan 22, 2025

• Add a3 ultra support

  samos123

  committed
  08d5a03

  Copy full SHA for 08d5a03

• update Dockerfile

  samos123

  committed
  e842543

  Copy full SHA for e842543

• remove original gpu dockerfile contents

  samos123

  committed
  565a187

  Copy full SHA for 565a187

Commits on Feb 5, 2025

• remove comments

  samos123

  committed
  2e01da0

  Copy full SHA for 2e01da0

• remove jax version override

  samos123

  committed
  56fe7c5

  Copy full SHA for 56fe7c5

• Merge branch 'main' into a3u-support

  samos123

  committed
  dc1e7cd

  Copy full SHA for dc1e7cd