google/flax

Pull Requests Commits

Copybara import of the project:

levskaya committed 3 years ago

6c0ebd9c

Merge pull request #2701 from cgarciae:cleanup-dense-general

a-googler committed 3 years ago

8a9fbc9c

Merge pull request #2557 from zaxtax:partial_serialization

a-googler committed 3 years ago

1d678b52

Merge pull request #2670 from google:mypy_build

a-googler committed 3 years ago

5bd4cd24

Merge pull request #2688 from IvyZX:pen

a-googler committed 3 years ago

f45efd37

Merge pull request #2541 from cgarciae:fix-property-error

a-googler committed 3 years ago

063f5414

Merge pull request #2697 from wookayin:drop-tf

a-googler committed 3 years ago

12e13846

Merge branch 'main' into pen

IvyZX committed 3 years ago

9e9c85f3

Introduce flax.io.NotFoundError to remove tensorflow dependency

wookayin committed 3 years ago

b0c2b9c4

Merge pull request #2702 from IvyZX:lrdep

a-googler committed 3 years ago

fd69b9e9

Merge pull request #2668 from IvyZX:flax-pen1

a-googler committed 3 years ago

12f2b27a

improve and migrate model_surgery guide

IvyZX committed 3 years ago

1adf0d8c

improve and migrate model_surgery guide

IvyZX committed 3 years ago

5dc65b9d

Merge pull request #2705 from IvyZX:push

a-googler committed 3 years ago

2946b3c8

IvyZX committed 3 years ago

2ea53884

Add deprecation message to flax.training.lr_schedule

IvyZX committed 3 years ago

5f8d876e

fix test_dense_general_batch_dim

cgarciae committed 3 years ago

bf60167e

revert test_dense_is_dense_general

cgarciae committed 3 years ago

ab34ff38

avoid splitting rng in DenseGeneral

cgarciae committed 3 years ago

f1160daa

arg-type ignore

zaxtax committed 3 years ago

e3afa838

zaxtax committed 3 years ago

8bd81696

Run mypy during testing

zaxtax committed 3 years ago

ecf76fa1

add mypy.ini placeholder.

levskaya committed 3 years ago

fec10eb6

Merge pull request #2682 from cgarciae:update-python-version

a-googler committed 3 years ago

4ba1d341

Merge pull request #2686 from chiamp:flax_docs_flax_basics

a-googler committed 3 years ago

51b34385

Update checkpointing guide with Orbax migration guide

IvyZX committed 3 years ago

a932ba79

zaxtax committed 3 years ago

93a51c79

Add serialization test for Partial

zaxtax committed 3 years ago

390b42c8

Add serialization for Partial functions. Completes #2433

zaxtax committed 3 years ago

6d2d804f

Merge pull request #2667 from chiamp:flax_docs

a-googler committed 3 years ago

09b6d569

Older