Code

Add JetStream to MaxText container#1058

Code

Open

DwarKapex wants to merge 6 commits intomainNVIDIA/JAX-Toolbox:mainfrom vkozlov/jetstream-4-maxtextNVIDIA/JAX-Toolbox:vkozlov/jetstream-4-maxtextCopy head branch name to clipboard

+57

Commits

Commits on Sep 23, 2024

• Add JetStream to MaxText container

  DwarKapex

  committed
  7517eba

Commits on Sep 25, 2024

• Merge branch 'main' into vkozlov/jetstream-4-maxtext

  DwarKapex

  authored
  333f231

• Sandbox test

  DwarKapex

  committed
  c952cb3

• Sandbox fix

  DwarKapex

  committed
  ddc4124

• Revert sandbox

  DwarKapex

  committed
  ab21dae

Commits on Oct 10, 2024

• Merge branch 'main' into vkozlov/jetstream-4-maxtext

  yhtang

  authored
  d9f5c18