google/flax

Pull Requests Commits

Internal-only change.

saeta committed 2 years ago

feba4152

Allow clipping the dynamic loss scale to a minimum value

a-googler committed 2 years ago

bb76e47c

Merge pull request #3058 from chiamp:readme

a-googler committed 2 years ago

12b0245d

linen_meta_test: make test compatible with JAX custom prng

Jake VanderPlas committed 2 years ago

2d47c931

Make flax tests compatible with jax.enable_custom_prng

Jake VanderPlas committed 2 years ago

fdf877cc

Merge pull request #3059 from Neilblaze:patch-fix-v2

a-googler committed 2 years ago

6916f4dd

updated var & corrected typo

Neilblaze committed 2 years ago

Verified 2a1bef4f

updated to constructor

Neilblaze committed 2 years ago

Verified 18a21855

fixed broken coverage link

chiamp committed 2 years ago

39a6656e

Merge pull request #3021 from Neilblaze:doc-patch-typo

a-googler committed 2 years ago

b3945f84

The pattern used in this change (`obj = object.__new__(data_clz)`) is not universally applicable to all Flax subclasses.

a-googler committed 2 years ago

dde03a18

fixed p-c hook trailing-whitespace

Neilblaze committed 2 years ago

Verified b48c765d

fixed notebooks

Neilblaze committed 2 years ago

Verified 823fd722

Merge pull request #3024 from google:optimize-pytreenode

a-googler committed 2 years ago

7c9b3b34

Merge pull request #2999 from google:add-bidirectional-docs

a-googler committed 2 years ago

d8708edd

optimize struct.dataclass

cgarciae committed 2 years ago

ab9051b6

add missing Bidirectional API reference

cgarciae committed 2 years ago

7e61fb1d

Avoid inadvertent submodule setup

chiamp committed 2 years ago

a648605b

Merge pull request #3039 from google:fix-release

a-googler committed 2 years ago

3a35b228

fix pythonpublish.yml

cgarciae committed 2 years ago

d0425186

Merge pull request #3035 from IvyZX:push

a-googler committed 2 years ago

6884e461

IvyZX committed 2 years ago

90eee947

Merge pull request #3029 from IvyZX:push

a-googler committed 2 years ago

31bd1f1a

Merge pull request #3031 from canyon289:patch-1

a-googler committed 2 years ago

92a9b40f

Update orbax_upgrade_guide.rst

canyon289 committed 2 years ago

Verified 01cf1cfd

Add more explicit warnings and logs for code with Orbax backend.

IvyZX committed 2 years ago

76dec35f

Back to orbax-checkpoint after the package is fixed in 0.2.1.

IvyZX committed 2 years ago

6a9e0806

Use backend dependent os.path.getsize.

lespeholt committed 2 years ago

0ac03c37

Merge pull request #3027 from IvyZX:test-fix

a-googler committed 2 years ago

a6be85d4

fix test and unbreak head

IvyZX committed 2 years ago

a517eee9

Older