Commits

History for

Commits on Apr 23, 2024

Pre-commit config
khatwanimohit
committedApr 23, 2024

Commits on Apr 9, 2024

Add functionality to automatically upload logs to Vertex Tensorboard
SurbhiJainUSC
committedApr 9, 2024

Commits on Apr 2, 2024

Maxtext Integration Goodput Package
dipannita08
committedApr 2, 2024

Commits on Mar 13, 2024

update orbax version to the latest
ssusie
committedMar 13, 2024

Commits on Mar 12, 2024

switch to orbax's new API and add an opttion to restore with SingleReplicaArrayHandler
ssusie
committedMar 12, 2024

Commits on Mar 11, 2024

Jetstream 1.0
Rafi Witten
committedMar 11, 2024
pin orbax to 5.3
ssusie
committedMar 11, 2024

Commits on Feb 13, 2024

[MLPerf GPT3] add c4_mlperf input pipeline and eval step
ZhiyuLi-goog
committedFeb 13, 2024

Commits on Feb 5, 2024

Copybara import of the project:

aireenmei
authored and
maxtex authors
committedFeb 5, 2024

Commits on Jan 31, 2024

Fix requirements.txt deps
khatwanimohit
committedJan 31, 2024

Commits on Jan 9, 2024

Simplify Jax Distributed System Initialization
Rafi Witten
committedJan 9, 2024

Commits on Jan 4, 2024

Set up pytype on CI tests
michelle-yooh
committedJan 4, 2024

Commits on Nov 16, 2023

move TF after tf-text to ensure a tf-text compatible TF is installed
Yash Shah
committedNov 16, 2023

Commits on Nov 11, 2023

Refactor to use AQT PIP package
gobbleturk
committedNov 11, 2023

Commits on Oct 27, 2023

Remove head mode from setup.sh (#225 )
shahyash10
committedOct 27, 2023
remove tbp nightly from setup.sh (#226 )
shahyash10
committedOct 27, 2023

Commits on Oct 4, 2023

Add support for installing nightly version oftensorboard_plugin_profile (#194 )
shahyash10
committedOct 4, 2023

Commits on Sep 27, 2023

Restricting TF to 2.13 to avoid extra installs.
Rafi Witten
committedSep 27, 2023

Commits on Jun 8, 2023

Add configuration to collect stack traces
SurbhiJainUSC
committedJun 8, 2023

Commits on May 12, 2023

Address the comments.
tonyjohnchen
committedMay 12, 2023

Commits on May 3, 2023

Reinstate Asynchronous checkpointer (#22 )
gobbleturk
committedMay 3, 2023

Commits on May 1, 2023

Call jax.device_count at start to solve TPU already registered issue (#21 )
gobbleturk
committedMay 1, 2023

Commits on Apr 19, 2023

Pin sentencepiece to 0.1.97
gobbleturk
committedApr 19, 2023

Commits on Apr 11, 2023

Merge pull request #98 from rwitten/rwitten_flax_normal_install
Rafi Witten
committedApr 11, 2023
Moved Flax to a normal install
Rafi Witten
committedApr 11, 2023

Commits on Mar 30, 2023

Pin orbax to 0.1.6 (#92 )
gobbleturk
committedMar 30, 2023

Commits on Mar 28, 2023

jax==0.4.6 on top in requirements (#89 )
khatwanimohit
committedMar 28, 2023

Commits on Mar 27, 2023

Pin protobuf to working version (#88 )
gobbleturk
committedMar 27, 2023

Commits on Mar 14, 2023

Pinned to jax==0.4.6 and updated results.
Rafi Witten
committedMar 14, 2023

Commits on Mar 7, 2023

Add profiler dependencies and remove compilation from profling (#73 )
bvandermoon
committedMar 7, 2023
Update requirements.txt
michelle-yooh
committedMar 7, 2023

Commits on Feb 8, 2023

Documentation Improvements
Rafi Witten
committedFeb 8, 2023

Commits on Jan 29, 2023

MXLA/PJRT is supported in public versions of Jax/Jaxlib/LibTPU so just using those.
Rafi Witten
committedJan 29, 2023

Commits on Jan 13, 2023

Change checkpointing to asynchronous (#18 )

gobbleturk
and
gobbleturk
committedJan 13, 2023

Commits on Dec 29, 2022

Updated dependency management.
Rafi Witten
committedDec 29, 2022