Code

Add JetStream to MaxText container#1058

Code

Open

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

+49-1

Commits

Commits on Sep 23, 2024

• Add JetStream to MaxText container

  DwarKapex

  committed
  7517eba

  Copy full SHA for 7517eba

Commits on Sep 25, 2024

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

  DwarKapex

  authored
  333f231

  Copy full SHA for 333f231

• Sandbox test

  DwarKapex

  committed
  c952cb3

  Copy full SHA for c952cb3

• Sandbox fix

  DwarKapex

  committed
  ddc4124

  Copy full SHA for ddc4124

• Revert sandbox

  DwarKapex

  committed
  ab21dae

  Copy full SHA for ab21dae

Commits on Oct 10, 2024

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

  yhtang

  authored
  d9f5c18

  Copy full SHA for d9f5c18

Commits on Jan 9, 2025

• Refactor MaxText docker and add upstream JetStream to the container

  DwarKapex

  committed
  9d59a81

  Copy full SHA for 9d59a81

• Remove second installation of JetStream

  DwarKapex

  committed
  9b7426a

  Copy full SHA for 9b7426a

Commits on Jan 10, 2025

• Refactor Dockerfile.maxtext

  DwarKapex

  committed
  07c5a54

  Copy full SHA for 07c5a54

• Merge main

  DwarKapex

  committed
  d726b1b

  Copy full SHA for d726b1b

• Add jetstream to manifest file, and updated main CI file

  DwarKapex

  committed
  43b2bd0

  Copy full SHA for 43b2bd0