google/flax

Pull Requests Commits

Add support for PyTrees with None leaves. Add additional unit tests.

cpgaffney1 committed 3 years ago

0c5232e6

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

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

Merge pull request #2667 from chiamp:flax_docs

a-googler committed 3 years ago

09b6d569

Merge pull request #2684 from chiamp:flax_docs_getting_started

a-googler committed 3 years ago

5c9ba92a

Update docs/guides/flax_basics.md

chiamp committed 3 years ago

7a15952e

Update docs/guides/flax_basics.md

chiamp committed 3 years ago

Verified 8b6f88cd

add descriptor wrapper

cgarciae committed 3 years ago

dd2f459f

Update docs/getting_started.md

chiamp committed 3 years ago

c0612752

Update docs/getting_started.md

chiamp committed 3 years ago

Verified e076859b

Merge pull request #2687 from chiamp:flax_docs_model_surgery

a-googler committed 3 years ago

db690a47

Merge pull request #2685 from chiamp:flax_docs_extracting_intermediates

a-googler committed 3 years ago

650dcb2b

Older