Commits

History for

Commits on May 16, 2024

Remove async XLA_FLAGS from A3 configs
michelle-yooh
committedMay 16, 2024

Commits on May 14, 2024

Remove cases where the deprecated --xla_gpu_simplify_all_fp_conversions is set to its default value.

dimitar-asenov
authored and
maxtext authors
committedMay 14, 2024

Commits on May 10, 2024

Add forward pass logit check test for Llama2-7b
A9isha
committedMay 10, 2024

Commits on May 3, 2024

Add more tests for Mixtral
RissyRan
committedMay 3, 2024
Add Llama2-70b test
A9isha
committedMay 3, 2024

Commits on May 1, 2024

Add tests to GPU runner
michelle-yooh
committedMay 1, 2024

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