Commits

Commits on May 17, 2024

Merge pull request #652 from google:yooh/xla-flags-fix
maxtext authors
committedMay 17, 2024

Commits on May 16, 2024

Add llama-70b gpu config.

golechwierowicz
authored and
maxtext authors
committedMay 16, 2024
Remove async XLA_FLAGS from A3 configs
michelle-yooh
committedMay 16, 2024

Commits on May 15, 2024

Merge pull request #649 from google:aireen/fix_pylint
maxtext authors
committedMay 15, 2024
Merge pull request #642 from google:yooh/streamline-unit-tests
maxtext authors
committedMay 15, 2024
fix pylint
aireenmei
committedMay 15, 2024
Merge pull request #645 from reedwm:xla_flags_fix
maxtext authors
committedMay 15, 2024
Merge branch 'main' into xla_flags_fix
reedwm
committedMay 15, 2024
streamline CI test structure
michelle-yooh
committedMay 15, 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
Merge pull request #640 from google:yooh/add_oss_tests
maxtext authors
committedMay 14, 2024

Commits on May 13, 2024

Eval the command string from XPK for GPU script
michelle-yooh
committedMay 13, 2024

Commits on May 10, 2024

Merge pull request #643 from google:anisha-debug-decoding
maxtext authors
committedMay 10, 2024
Add forward pass logit check test for Llama2-7b
A9isha
committedMay 10, 2024
Merge pull request #641 from google:prefill-optimize
maxtext authors
committedMay 10, 2024
Merge pull request #644 from google:moe_param
maxtext authors
committedMay 10, 2024
Update llama2_7b_gpu.yml
maxtext authors
committedMay 10, 2024
Remove async XLA_FLAGS from A3 configs.
reedwm
committedMay 10, 2024

Commits on May 9, 2024

fix sharding on generate cache in prefill results.
jwyang-google
committedMay 9, 2024
Update tflops calculation
RissyRan
committedMay 9, 2024

Commits on May 7, 2024

Merge pull request #635 from google:mattdavidow-numpy-prod
maxtext authors
committedMay 7, 2024
Reverts e8b53e5

golechwierowicz
authored and
maxtext authors
committedMay 7, 2024

Commits on May 6, 2024

Make some AQT dataclasses to use keyword-only fields (1/N)
maxtext authors
committedMay 6, 2024

Commits on May 3, 2024

Merge pull request #627 from google:ranran_more_moe
maxtext authors
committedMay 3, 2024
Add more tests for Mixtral
RissyRan
committedMay 3, 2024
Internal change only.
maxtext authors
committedMay 3, 2024
Merge pull request #639 from google:anisha-llama2-70b-test
maxtext authors
committedMay 3, 2024
Add Llama2-70b test
A9isha
committedMay 3, 2024

Commits on May 2, 2024

Merge pull request #637 from google:lizhiyu/miss_sharding
maxtext authors
committedMay 2, 2024
fix norm sharding
ZhiyuLi-goog
committedMay 2, 2024

Commits on May 1, 2024

Merge pull request #630 from google:yooh/gpu-unit-tests
maxtext authors
committedMay 1, 2024
Replace deprecated np.product with np.prod
gobbleturk
committedMay 1, 2024
Add tests to GPU runner
michelle-yooh
committedMay 1, 2024

Commits on Apr 30, 2024

Merge pull request #633 from google:rwitten_setup_streamlined
maxtext authors
committedApr 30, 2024
Merge branch 'main' into rwitten_setup_streamlined
rwitten
committedApr 30, 2024