Code

Input batch sharding strategy BATCH#884

Code

Open

apoorvtintin wants to merge 1 commit intoapple:mainapple/axlearn:mainfrom patrick-toulme:mainline_upstream_input_shardingpatrick-toulme/axlearn:mainline_upstream_input_shardingCopy head branch name to clipboard

+87-31

Commits

Commits on Dec 11, 2024

• Input batch sharding strategy BATCH

  apoorvtintin

  committed
  056536d

  Copy full SHA for 056536d