google/flax

Pull Requests Commits

Merge branch 'google:main' into update-how-to-build-flax-docs

8bitmp3 committed 3 years ago

Verified d501c6e4

Merge pull request #2730 from IvyZX:pjit

a-googler committed 3 years ago

0453d38f

Update How to Build Flax Docs in README

8bitmp3 committed 3 years ago

6f76a272

pjit guide final touches

IvyZX committed 3 years ago

4920bd85

Update Flax pjit guide notebook

8bitmp3 committed 3 years ago

Verified dcf5517e

create pjit spmd guide doc

IvyZX committed 3 years ago

023511b6

Merge pull request #2691 from chiamp:dict_serialization_errors

a-googler committed 3 years ago

3acb0e0a

Merge pull request #2734 from cgarciae:simplify-axis-rules

a-googler committed 3 years ago

8278a404

Merge pull request #2747 from 8bitmp3:update-batchnorm-title

a-googler committed 3 years ago

002a63fb

Update Flax BatchNorm guide title

8bitmp3 committed 3 years ago

Verified db4cf176

Merge pull request #2746 from google:docs-38

a-googler committed 3 years ago

25bf6470

Merge pull request #2675 from 8bitmp3:dropout-guide

a-googler committed 3 years ago

d6591450

Update readthedocs to python 3.8

IvyZX committed 3 years ago

Verified c6cbb430

Merge branch 'main' of https://www.github.com/google/flax into scope-tests

marcvanzee committed 3 years ago

05606274

Adds some more comments to scope.py

marcvanzee committed 3 years ago

119335ed

Merge branch 'google:main' into update-how-to-build-flax-docs

8bitmp3 committed 3 years ago

Verified 262ac632

Update How to Build Flax Docs in README

8bitmp3 committed 3 years ago

7b0bc669

Monitor checkpoint load/save durations.

a-googler committed 3 years ago

4960e96a

Fix some tiny inconsistencies between scope.py and module.py.

marcvanzee committed 3 years ago

7cbbe0cd

Add Flax Dropout guide

8bitmp3 committed 3 years ago

cc390789

make _AxisRules a threading.local object

cgarciae committed 3 years ago

ff8c4724

Merge pull request #2732 from levskaya:refixci

a-googler committed 3 years ago

bf45e352

Re-enable faster github CI runners

levskaya committed 3 years ago

34b35835

Support multi output layers in the Sequential combinator

aboSamoor committed 3 years ago

6d5bc2a8

Merge pull request #2729 from levskaya:queue_runner_fix

a-googler committed 3 years ago

4f249333

Revert to using public github runner pool while internal pool issues are

levskaya committed 3 years ago

eb9cecf9

Merge pull request #2680 from marcvanzee:merge_param

a-googler committed 3 years ago

4669ea04

Merge pull request #2726 from google:marcvanzee-patch-13

a-googler committed 3 years ago

51018066

Merge pull request #2717 from marcvanzee:import-inits

a-googler committed 3 years ago

d0b09b6e

Bump action/setup-python Python version to 3.9

marcvanzee committed 3 years ago

6f34791b

Newer Older