google/flax

Pull Requests Commits

Merge pull request #2028 from levskaya:getattribute

a-googler committed 3 years ago

dba3fa59

Add __getattribute__ with lazy setup trigger.

levskaya committed 3 years ago

aca3e29d

Merge pull request #2025 from jheek:fix-tranform-state-reuse

a-googler committed 3 years ago

0c647cbb

Merge pull request #2026 from jheek:remove-more-tree_multimap

a-googler committed 3 years ago

ffba15be

Remove tree_multimap from docs, examples, and tests

jheek committed 3 years ago

c2a62129

Fix bug where the linen Module state is reused.

jheek committed 3 years ago

8c6a16bf

Merge pull request #2024 from jheek:remove-tree_multimap-calls

a-googler committed 3 years ago

04617455

Use tree_map instead of deprecated tree_multimap

jheek committed 3 years ago

f7947da3

Merge pull request #1990 from lkhphuc:phuc-mypy

a-googler committed 3 years ago

6ab8cb27

Merge branch 'main' into phuc-mypy

levskaya committed 3 years ago

Verified 869e6505

Merge pull request #2020 from jheek:lifted-cond

a-googler committed 3 years ago

cd5c4d7c

Merge pull request #2015 from melissatan:intro

a-googler committed 3 years ago

1cb69359

Update Jax and Flax intro linear regression examples.

melissatan committed 3 years ago

8ff11aa7

Add lifted cond

jheek committed 3 years ago

d2448870

Unpin pytype dependency and fix various pytype/typing errors.

levskaya committed 3 years ago

60578551

Make init_fn optional for variable

jheek committed 3 years ago

12427842

Clarifies `optim.Adam(weight_decay)` parameter.

andsteing committed 3 years ago

b4e9b12f

Merge pull request #2018 from google:levskaya-docs-tweak-1

a-googler committed 3 years ago

85ac2ca5

Merge pull request #1927 from NeilGirdhar:dataclass_field

a-googler committed 3 years ago

313014ea

Don't recommend mixing setup and compact in docs.

levskaya committed 3 years ago

Verified 8bed5714

Merge pull request #2013 from melissatan:docs

a-googler committed 3 years ago

c170b26d

Update ScopeParamNotFoundError message and docstring example.

melissatan committed 3 years ago

b5087ff9

Merge pull request #2009 from jheek:add-padding-canonicalize

a-googler committed 3 years ago

39038383

Merge pull request #2012 from andsteing:imagenet2

a-googler committed 3 years ago

f35d5576

Bumps flax/jax versions in examples/imagenet.

andsteing committed 3 years ago

2e315cff

Merge pull request #2010 from andsteing:release

a-googler committed 3 years ago

ad8dfe01

Bumps version to 0.4.1

andsteing committed 3 years ago

0377fd4e

Canonicalize conv padding

jheek committed 3 years ago

2e620b34

Merge pull request #1978 from melissatan:docs

a-googler committed 3 years ago

e16cf72b

Merge pull request #2007 from jheek:fix-non-string-keys-in-statedict

a-googler committed 3 years ago

049096b9

Newer Older