google/flax

Pull Requests Commits

Allow Adafactor to not update certain parameters proportional

a-googler committed 4 years ago

8a090e3e

Merge pull request #1242 from andsteing/docker

andsteing committed 4 years ago

Verified 4a0e32e5

Fix Linen Module with empty containers in the attributes values.

jpuigcerver committed 4 years ago

862cc828

Merge pull request #1182 from marcvanzee/rtd-training

marcvanzee committed 4 years ago

Verified 593e53bb

Merge pull request #1182 from marcvanzee:rtd-training

a-googler committed 4 years ago

2b2acf1a

Merge pull request #1237 from jheek:fix-dense-general

a-googler committed 4 years ago

f19a720a

Merge pull request #1243 from andsteing/flip

andsteing committed 4 years ago

Verified f704d531

Merge branch 'master' of https://github.com/google/flax into rtd-training

marcvanzee committed 4 years ago

5a2e9c4b

Add test for multiple non final axes

jheek committed 4 years ago

c558a7c3

Adds FLIPs link to main RST nav.

andsteing committed 4 years ago

688f0b5a

Git ignores .devcontainer

andsteing committed 4 years ago

81b398bb

Fix DenseGeneral bug:

jheek committed 4 years ago

0f9acdc8

Merge pull request #1236 from jheek:fix-tuple-type

a-googler committed 4 years ago

767a3e94

Copybara import of the project:

gnecula committed 4 years ago

f31023b4

Merge pull request #711 from gmittal:master

a-googler committed 4 years ago

6faf6c1a

fix wrong tuple type in Scope constructor.

jheek committed 4 years ago

2a719936

Merge branches 'master' and 'master' of https://github.com/google/flax into rtd-training

marcvanzee committed 4 years ago

9a41e049

Allow Flax lifted transforms to work on partially applied Modules.

levskaya committed 4 years ago

c2e85837

Merge pull request #1210 from davisyoshida:annotation_fix

a-googler committed 4 years ago

d4102ff6

Added missing @ causing functools.wrap to have no effect in dynamic_scale.py

davisyoshida committed 4 years ago

d03a2622

Merge pull request #1202 from lucasb-eyer:patch-6

a-googler committed 4 years ago

111e30e0

Merge pull request #1200 from hmph:checkpoint-fix

a-googler committed 4 years ago

95409ece

Performance improvements for ImageNet example:

jheek committed 4 years ago

266052cb

Merge pull request #1203 from jheek:recurrent-setup

a-googler committed 4 years ago

a72c42ce

Don't wrap user-defined __post_init__ function.

levskaya committed 4 years ago

bc31315c

Merge branch 'master' of https://github.com/google/flax into rtd-training

marcvanzee committed 4 years ago

160b7529

Merge pull request #1204 from jheek/bugfix-in-pr-template

marcvanzee committed 4 years ago

Verified 7fff536b

Allow a small bugfix to be send as a PR straight away.

jheek committed 4 years ago

05a1b692

Fix super calls in compact & setup

jheek committed 4 years ago

c31388de

Ask dataclass to generate __hash__ only if not implemented yet.

lucasb-eyer committed 4 years ago

dd8b4790

Older