Code

TRN2 Meshes and Configurations#916

Code

Open

apoorvtintin wants to merge 9 commits intoapple:mainapple/axlearn:mainfrom apoorvtintin:mainline-upstream-boilerplateapoorvtintin/axlearn:mainline-upstream-boilerplateCopy head branch name to clipboard

+4,437-17

Commits

Commits on Jan 31, 2025

• TRN2 Meshes and Configurations

  apoorvtintin

  committed
  17c7d66

  Copy full SHA for 17c7d66

• Add get_recursive and set_recursive to ConfigBase.

  apoorvtintin

  committed
  71e006a

  Copy full SHA for 71e006a

Commits on Feb 3, 2025

• Use loops inside get/set_recursively

  Show description for 8d976db
  apoorvtintin

  committed
  8d976db

  Copy full SHA for 8d976db

Commits on Feb 5, 2025

• Update partition spec

  apoorvtintin

  committed
  4f5db93

  Copy full SHA for 4f5db93

• Use get_recursively inside set

  apoorvtintin

  committed
  78fdd29

  Copy full SHA for 78fdd29

Commits on Feb 6, 2025

• Move trn2 configs to a helper function.

  Show description for 4613fba
  apoorvtintin

  committed
  4613fba

  Copy full SHA for 4613fba

• TRN2 partitionspec supports DP over FSDP and TP

  apoorvtintin

  committed
  38f8111

  Copy full SHA for 38f8111

Commits on Feb 10, 2025

• Use for loop in get_recursively

  apoorvtintin

  committed
  2b1dad3

  Copy full SHA for 2b1dad3

• Update Golden Configs

  apoorvtintin

  committed
  53472f2

  Copy full SHA for 53472f2