Code

Add support for axlearn#1285

Code

Open

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

+296

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