8000 Commits · Bloodsolz/maxtext · GitHub

More Web Proxy on the site http://driver.im/

Commits

Add training tests to Mistral and Mixtral

committed

Merge pull request #487 from google:anisha-fix-llama2-test

maxtext authors

committed

Commits on Mar 7, 2024

update end_to_end/test_llama2.sh to include checkpoint conversion in the test
A9isha
committed
Merge pull request #492 from google:mattdavidow-fix-head-post-copybara
maxtext authors
committed
Use flash attention for llama2_7b_single_host config

cheshire
authored and
michelle-yooh
committed
Config for benchmarking single-host 7b llama2.

cheshire
authored and
maxtext authors
committed
Merge pull request #485 from google:aireen/grain_unit_test2
maxtext authors
committed
Merge pull request #489 from google:lizhiyu/add_remat_policy
maxtext authors
committed
Merge pull request #491 from google:aireen/fix_setup
maxtext authors
committed
Copybara import of the project:

A9isha
authored and
maxtext authors
committed
Fix head post g3 push
gobbleturk
committed
Add basic inheritance for YAML config files using base_config attribute.

cheshire
authored and
maxtext authors
committed

Commits on Mar 6, 2024

Commits on Mar 4, 2024

Merge pull request #466 from google:ansiha-ckpt-perf-cpu
maxtext authors
committed
Merge pull request #470 from google:mattdavidow-fix-dtype
maxtext authors
committed
shard checkpoint if applicable
A9isha
committed
Fix weight dtypes to float32
gobbleturk
committed
Merge pull request #476 from google:msingh-aqt-rm
maxtext authors
committed

Commits on Mar 3, 2024

Removing param values for which we store aqt tensors to reduce memory usage.
mitalisi
committed

Commits on Mar 1, 2024

add unit test for grain ckpt
aireenmei
committed

Commits on Feb 29, 2024

Commits on Feb 28, 2024

Install jax-nightly for MODE=nightly in setup.sh
SurbhiJainUSC
committed

0