Commits

History for

Commits on Apr 30, 2024

Streamlined setup.sh to have fewer apt install calls
Rafi Witten
committedApr 30, 2024

Commits on Apr 23, 2024

Pre-commit config
khatwanimohit
committedApr 23, 2024

Commits on Apr 17, 2024

Move apt install from rto_setup.sh to setup.sh
tonyjohnchen
committedApr 17, 2024

Commits on Apr 5, 2024

Update TE version
chajath
committedApr 5, 2024

Commits on Apr 1, 2024

Activate version pinning for pinned mode only
chajath
committedApr 1, 2024

Commits on Mar 25, 2024

Prebuild and install Transformer Engine package
chajath
committedMar 25, 2024

Commits on Mar 19, 2024

Added pedagogical example/shamp_collective_matmul.py
Rafi Witten
committedMar 19, 2024

Commits on Mar 9, 2024

Update setup.sh
eltociear
committedMar 9, 2024

Commits on Mar 8, 2024

Update setup.sh
eltociear
committedMar 8, 2024

Commits on Mar 7, 2024

Merge pull request #491 from google:aireen/fix_setup
maxtext authors
committedMar 7, 2024

Commits on Mar 6, 2024

use apt-key for installing gcsfuse
aireenmei
committedMar 6, 2024

Commits on Mar 5, 2024

build and upload gpu docker images
NinaCai
committedMar 5, 2024
Remove head mode for setup.sh since we don't use this mode anymore.
tonyjohnchen
committedMar 5, 2024
Merge pull request #467 from google:aireen/grain_unit_test
maxtext authors
committedMar 5, 2024

Commits on Mar 1, 2024

add unit test for grain ckpt
aireenmei
committedMar 1, 2024

Commits on Feb 28, 2024

Install jax-nightly for MODE=nightly in setup.sh
SurbhiJainUSC
committedFeb 28, 2024

Commits on Feb 23, 2024

adding gcloud sdk in dockerfile for gpu and tpu
ssusie
committedFeb 23, 2024

Commits on Feb 21, 2024

Replace the base of GPU image with NVIDA jax-toolbox image
michelle-yooh
committedFeb 21, 2024

Commits on Feb 20, 2024

Move all dependencies from preflight.sh to setup.sh
tonyjohnchen
committedFeb 20, 2024

Commits on Jan 31, 2024

Fix requirements.txt deps
khatwanimohit
committedJan 31, 2024

Commits on Jan 5, 2024

Fix build commands in setup.sh for head mode
SurbhiJainUSC
committedJan 5, 2024

Commits on Nov 1, 2023

Merge branch 'main' into abhinavgoel95-patch-1
abhinavgoel95
committedNov 1, 2023

Commits on Oct 27, 2023

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

Commits on Oct 20, 2023

Update setup.sh to set default device to tpu
abhinavgoel95
committedOct 20, 2023

Commits on Oct 19, 2023

Exiting with exit 1

abhinavgoel95
authored and
khatwanimohit
committedOct 19, 2023
graceful exit for unsupport configs

abhinavgoel95
authored and
khatwanimohit
committedOct 19, 2023
Adding support for GPU

abhinavgoel95
authored and
khatwanimohit
committedOct 19, 2023

Commits on Oct 4, 2023

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

Commits on Sep 8, 2023

2 small fixes.
tonyjohnchen
committedSep 8, 2023
Make sure to pass through dependencies when building.
Rafi Witten
committedSep 8, 2023

Commits on Aug 8, 2023

Retry setup.sh
gobbleturk
committedAug 8, 2023

Commits on Jul 21, 2023

Remove all sudo
tonyjohnchen
committedJul 21, 2023

Commits on Jul 7, 2023

Add version option and custom LIBTPU to stable (#82 )
gobbleturk
committedJul 7, 2023

Commits on May 16, 2023

Update the error message.
tonyjohnchen
committedMay 16, 2023