Commits

History for

Commits on Apr 25, 2024

Split Mixtral test into two scripts
RissyRan
committedApr 25, 2024

Commits on Apr 23, 2024

Pre-commit config
khatwanimohit
committedApr 23, 2024

Commits on Apr 12, 2024

Share GCS path between Gemma-7b tests
A9isha
committedApr 12, 2024

Commits on Apr 11, 2024

Move tpu end-to-end test scripts to tpu folder
NinaCai
committedApr 11, 2024

Commits on Apr 10, 2024

Separate gpu and tpu end-to-end scripts
NinaCai
committedApr 10, 2024

Commits on Apr 9, 2024

Fix Gemma links
khatwanimohit
committedApr 9, 2024

Commits on Apr 2, 2024

Streamline params usage
anfals
committedApr 2, 2024
Test decode and finetuning for Gemma
khatwanimohit
committedApr 2, 2024

Commits on Mar 27, 2024

Revert "Make usage of params consistent"
khatwanimohit
committedMar 27, 2024

Commits on Mar 26, 2024

Make usage of params consistent
anfals
committedMar 26, 2024

Commits on Mar 25, 2024

Update Llama2 readme and test
A9isha
committedMar 25, 2024

Commits on Mar 22, 2024

Revert "Increase test name entropy"
jonb377
committedMar 22, 2024

Commits on Mar 15, 2024

Increase test name entropy
jonb377
committedMar 15, 2024

Commits on Mar 14, 2024

Copybara import of the project:

RissyRan
authored and
maxtext authors
committedMar 14, 2024

Commits on Mar 13, 2024

Merge pull request #515 from google:rwitten_unroll_all_checkpoints
maxtext authors
committedMar 13, 2024
Allow support for generate param only checkpoint to do param to param
Rafi Witten
committedMar 13, 2024
change default folder to items for ckpt in new tests
ssusie
committedMar 13, 2024
Modify convergence test to read in run_name from env
raymondzouu
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

Commits on Mar 8, 2024

Add training tests to Mistral and Mixtral
RissyRan
committedMar 8, 2024

Commits on Mar 7, 2024

update end_to_end/test_llama2.sh to include checkpoint conversion in the test
A9isha
committedMar 7, 2024
Merge pull request #485 from google:aireen/grain_unit_test2
maxtext authors
committedMar 7, 2024
Copybara import of the project:

A9isha
authored and
maxtext authors
committedMar 7, 2024

Commits on Mar 5, 2024

add unit test for checkpoint compatibility between tfds and grain
aireenmei
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 29, 2024

add option to shorten convergence test steps
gobbleturk
committedFeb 29, 2024

Commits on Feb 23, 2024

Fix quotes in metrics filename
gobbleturk
committedFeb 23, 2024

Commits on Feb 22, 2024

Merge pull request #440 from google:mohit/gemma_tokenizer
maxtext authors
committedFeb 22, 2024
refactor maxtext checkpointing and integrate with grain
aireenmei
committedFeb 22, 2024

Commits on Feb 21, 2024

Add Llama2/Mistral/Mixtral test instructions for external users
RissyRan
committedFeb 21, 2024
add gemma tokenizer
khatwanimohit
committedFeb 21, 2024
Introduce model Gemma
khatwanimohit
committedFeb 21, 2024

Commits on Feb 19, 2024

Revert "Merge pull request #416 from google:aireen/grain_checkpointing"
Rafi Witten
committedFeb 19, 2024