Search results

Refactor profiler in trainers

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1833

#1833 In AI-Hypercomputer/maxtext;

· SurbhiJainUSC opened on Jun 14, 2025

Add total memory usage logging in GB and unit tests for activation offload

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1813

#1813 In AI-Hypercomputer/maxtext;

· zhenying-liu opened on Jun 10, 2025

Run code style across repo

pull ready

Status: Draft (not ready).

AI-Hypercomputer/maxtextnumber 1639

#1639 In AI-Hypercomputer/maxtext;

· richjames0 opened on Apr 25, 2025

Moving the checkpointing util funtions to a spearate file to avoid circular dependency

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1623

#1623 In AI-Hypercomputer/maxtext;

· wang2yn84 opened on Apr 24, 2025

Adding a warning when user tries to load train.py in the old manner (as a script)

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1620

#1620 In AI-Hypercomputer/maxtext;

· richjames0 opened on Apr 23, 2025

Update the megablox kernel utility function to support float16

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1423

#1423 In AI-Hypercomputer/maxtext;

· wang2yn84 opened on Mar 19, 2025

script to run mlperf llama2 70b on gpu

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1373

#1373 In AI-Hypercomputer/maxtext;

· jwyang-google opened on Mar 10, 2025

Add Assert in runner to avoid invalid gs bucket

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1367

#1367 In AI-Hypercomputer/maxtext;

· Obliviour opened on Mar 8, 2025

Link to maxtext_xpk_runner in readme for getting started with XPK

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1275

#1275 In AI-Hypercomputer/maxtext;

· Obliviour opened on Feb 14, 2025

Maxtext unit tests with Pathways backend.

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1211

#1211 In AI-Hypercomputer/maxtext;

· RoshaniN opened on Jan 28, 2025

Adding back building images with mode=stable

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1173

#1173 In AI-Hypercomputer/maxtext;

· parambole opened on Jan 16, 2025

Incorporate Orbax emergency replicator checkpoint manager

pull ready

Status: Open (in progress).

AI-Hypercomputer/maxtextnumber 1153

#1153 In AI-Hypercomputer/maxtext;

· xuefgu opened on Jan 8, 2025