Code

Add support for axlearn#1285

Code

Open

Steboss wants to merge 11 commits intomainNVIDIA/JAX-Toolbox:mainfrom sbosisio/support_axlearnNVIDIA/JAX-Toolbox:sbosisio/support_axlearnCopy head branch name to clipboard

+844-557

Commits

Commits on Jan 31, 2025

• start drafting support for axlearn

  Steboss

  committed
  5bea679

  Copy full SHA for 5bea679

Commits on Feb 3, 2025

• fix test for axlearn

  Steboss

  committed
  807d3df

  Copy full SHA for 807d3df

• add build for axlearn

  Steboss

  committed
  9947c08

  Copy full SHA for 9947c08

Commits on Feb 4, 2025

• install dependencies

  Steboss

  committed
  ef775d5

  Copy full SHA for ef775d5

• check tests

  Steboss

  committed
  15927d2

  Copy full SHA for 15927d2

Commits on Feb 6, 2025

• make the bash script executable

  Steboss

  committed
  af9dad3

  Copy full SHA for af9dad3

Commits on Feb 7, 2025

• minimal ci to test axlearn

  Steboss

  committed
  9e4d4a5

  Copy full SHA for 9e4d4a5

Commits on Feb 10, 2025

• fix requirements

  Steboss

  committed
  a3b8f26

  Copy full SHA for a3b8f26

• fix installation from pip

  Steboss

  committed
  94054fd

  Copy full SHA for 94054fd

• remove the nvidia-cunn package

  Steboss

  committed
  b9e893c

  Copy full SHA for b9e893c

• fix input for tests

  Steboss

  committed
  4088e2f

  Copy full SHA for 4088e2f