google/flax

Pull Requests Commits

updated flax docs

chiamp committed 3 years ago

c5aa714a

updated model_surgery in flax_docs

chiamp committed 3 years ago

33f95f8d

updated flax_basics

chiamp committed 3 years ago

d18be552

updated extracting_intermediates in flax docs

chiamp committed 3 years ago

f00cadc4

updated getting_started in flax docs

chiamp committed 3 years ago

291f1263

Merge pull request #2676 from IvyZX:misc2

a-googler committed 3 years ago

768c64dc

ignore disutils DeprecationWarning

cgarciae committed 3 years ago

60ad79f4

Check file or directory before removing checkpoints.

IvyZX committed 3 years ago

7513590c

update python version support

cgarciae committed 3 years ago

a8832a81

Merge pull request #2625 from chiamp:flax_io_warning

a-googler committed 3 years ago

0dc29e27

Merge pull request #2657 from chiamp:serialization_float128

a-googler committed 3 years ago

dbef7c45

Implements metadata FLIP

levskaya committed 3 years ago

7563d8fe

Merge pull request #2665 from j-towns:fix-DenseGeneral-rng

a-googler committed 3 years ago

b83875fc

Added check for Mac M1 chip, when conducting serialization test.

chiamp committed 3 years ago

4c3268c6

update docstrings and error messages in traverse_util.py.

IvyZX committed 3 years ago

d19cd260

Merge pull request #2595 from cgarciae:improve-transfer-learning-guide

a-googler committed 3 years ago

47dddc49

Merge pull request #2591 from cgarciae:n2590

a-googler committed 3 years ago

3604224e

improve transfer learning guide

cgarciae committed 3 years ago

c3f73e81

Updated flax docs.

chiamp committed 3 years ago

2593b03b

Updated flax documentation.

chiamp committed 3 years ago

80496515

Updated flax documentation.

chiamp committed 3 years ago

3c0d04e4

Some minor fixes and docstring updates following the last Orbax checkpoint migration PR.

IvyZX committed 3 years ago

3af21539

Merge pull request #2536 from cgarciae:batch-norm-guide

a-googler committed 3 years ago

d291af9a

Use a different rng key for each batch element in DenseGeneral

j-towns committed 3 years ago

09358e60

Fix train_state.TrainState.create rendering in Flax BatchNorm guide

8bitmp3 committed 3 years ago

Verified de58b2a2

Update Flax BatchNorm guide (fix formatting)

8bitmp3 committed 3 years ago

Verified 0e5baa89

Update and lint Flax BatchNorm guide

8bitmp3 committed 3 years ago

Verified 37c686ae

Merge pull request #2654 from cgarciae:n2636

a-googler committed 3 years ago

acb4ce05

Remove extra deserialization tests due to some observed breakages.

levskaya committed 3 years ago

bec978bb

Merge pull request #2632 from chiamp:serialization_errors

a-googler committed 3 years ago

a046c3c2

Newer Older