Commits

History for

Commits on Oct 31, 2023

Sync with internal changes

maxtext authors
authored and
michelle-yooh
committedOct 31, 2023

Commits on Oct 4, 2023

Use jax.jit in Maxtext and remove pjit by converting PartitionSpec to NamedShardings.
michelle-yooh
committedOct 4, 2023

Commits on Sep 20, 2023

comments
khatwanimohit
committedSep 20, 2023

Commits on Sep 19, 2023

fix shardings profiler
khatwanimohit
committedSep 19, 2023

Commits on Sep 18, 2023

fix ici_fsdp_parallelism (#161 )
yashs97
committedSep 18, 2023

Commits on Jul 25, 2023

Replace in_axis_resources with in_shardings (#96 )
gobbleturk
committedJul 25, 2023

Commits on Jun 20, 2023

Avoid deprecated imports
Rafi Witten
committedJun 20, 2023

Commits on May 26, 2023

Merge pull request #44 from google/update-import
tonyjohnchen
committedMay 26, 2023

Commits on May 25, 2023

Update import since partition_spec and Mesh is no longer in experimental folder.
tonyjohnchen
committedMay 25, 2023

Commits on May 11, 2023

Clean up remaining instances of pjrt and small grammer in README.
Obliviour
committedMay 11, 2023

Commits on Mar 28, 2023

Adding License, CONTRIBUTING.md, fixing pylintrc (#84 )
khatwanimohit
committedMar 28, 2023

Commits on Mar 19, 2023

Change default embedding dimension and batch size
michelle-yooh
committedMar 19, 2023

Commits on Mar 7, 2023

Add profiler dependencies and remove compilation from profling (#73 )
bvandermoon
committedMar 7, 2023

Commits on Mar 2, 2023

Support multi-sharding option in sharding script
michelle-yooh
committedMar 2, 2023

Commits on Feb 17, 2023

Add argparse to process inputs to the script
michelle-yooh
committedFeb 17, 2023

Commits on Feb 9, 2023

Profiler in Shardings.py
Rafi Witten
committedFeb 9, 2023

Commits on Jan 30, 2023

Added an example that shows different shardings
Rafi Witten
committedJan 30, 2023

Loading