google/flax

Pull Requests Commits

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

replace segmentation_mask in favor of seq_lengths

cgarciae committed 2 years ago

aa3774ca

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

Bump tensorflow from 2.9.3 to 2.11.1 in /examples/ogbg_molpcba

dependabot[bot] committed 2 years ago

Verified ad621f8e

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

Merge pull request #3002 from IvyZX:orbax-upgrade-guide

a-googler committed 2 years ago

8f55865b

Update Flax/Orbax save and load checkpoints

IvyZX committed 2 years ago

6b07e09f

Merge pull request #3022 from chiamp:pretty_repr

a-googler committed 2 years ago

660fd6c1

Added pretty_repr utility fn

chiamp committed 2 years ago

f55c3bf1

Merge pull request #3020 from Mddct:patch-1

a-googler committed 2 years ago

733423a2

Merge pull request #3012 from google:pyproject.toml-setup

a-googler committed 2 years ago

03e738ac

Neilblaze committed 2 years ago

Verified fb4d42bb

Newer Older