google/flax

Pull Requests Commits

Enable support for non-string keys in state dicts.

jheek committed 3 years ago

6226a9cc

Add constant initializer.

jpuigcerver committed 3 years ago

08f4c53d

Fix linter errors in transforms.py and improves some docstrings.

marcvanzee committed 3 years ago

9bd65b20

Merge pull request #2001 from levskaya:alias_pytree_bug

a-googler committed 3 years ago

5036e135

Fix remaining setup aliasing conflict edgecase.

levskaya committed 3 years ago

0658c0b2

Merge pull request #1996 from google:marcvanzee-patch-13

a-googler committed 3 years ago

3f068890

Fix linter errors in and improves some docstrings in flax.core.

marcvanzee committed 3 years ago

08984371

Merge pull request #1995 from GeoffNN:patch-1

a-googler committed 3 years ago

5a5aa87a

Fix typo in errors.py

marcvanzee committed 3 years ago

Verified 6875a531

Add docstring to SelfAttention module, specifying call args.

GeoffNN committed 3 years ago

Verified d5f0fa6c

Merge pull request #1994 from sudhakarsingh27:update-groupnorm-error-msg

a-googler committed 3 years ago

801b994d

Merge pull request #1961 from marcvanzee:upgrade-guide

a-googler committed 3 years ago

81fbbe56

Merge pull request #1983 from marcvanzee:howto-colab

a-googler committed 3 years ago

1872acb7

Merge pull request #1240 from google:max-commits

a-googler committed 3 years ago

4becb565

Rename state params howto

marcvanzee committed 3 years ago

d98d5b35

Merge branch 'main' of https://www.github.com/google/flax into howto-colab

marcvanzee committed 3 years ago

3484cda9

Update GroupNorm error msg to account for default value of num_groups

sudhakarsingh27 committed 3 years ago

572c9333

Merge pull request #1984 from marcvanzee:fix-sphinx-require

a-googler committed 3 years ago

ace68e2e

explicit export module in __init__.py

lkhphuc committed 3 years ago

ee424363

Merge pull request #1985 from google:dependabot/pip/examples/vae/pillow-9.0.1

a-googler committed 3 years ago

abd6b79b

Bump pillow from 9.0.0 to 9.0.1 in /examples/vae

dependabot[bot] committed 3 years ago

Verified 65aaee3e

Moves state params howto

marcvanzee committed 3 years ago

e3f191b8

Adds ipython_genutils to doc requirements

marcvanzee committed 3 years ago

b29a99e1

Cleans up Colab links

marcvanzee committed 3 years ago

e5bd7961

Fix linter errors in recurrent.py and stochastic.py.

marcvanzee committed 3 years ago

571018d1

Merge pull request #1971 from jheek:fix-transform-invalid-target-error

a-googler committed 3 years ago

59b7878e

Merge branch 'main' into docs

melissatan committed 3 years ago

Verified 5f8c5f93

Include activations in Sequential example.

melissatan committed 3 years ago

02c74c52

Fix linter errors in normalization.py.

marcvanzee committed 3 years ago

da5e204a

Merge branch 'main' into dataclass_field

levskaya committed 3 years ago

Verified 879ad876

Newer Older