google/flax

Pull Requests Commits

Merge pull request #417 from levskaya:linen3

a-googler committed 5 years ago

14883b01

Remove obsolete python3 linting directives.

levskaya committed 5 years ago

f8fac0ea

remove tie_in from linen

levskaya committed 5 years ago

979a5b98

Fix nits in the last commit.

mohitreddy1996 committed 5 years ago

09385941

Introduce unit test for sst2 training and evaluation loop by running it on a single epoch with 8 examples.

mohitreddy1996 committed 5 years ago

ac3073d1

Add automatic named call wrapping for profiling.

levskaya committed 5 years ago

c82fa6da

Implement multi-scope transforms for linen.

levskaya committed 5 years ago

c3446638

Don't automatically attach "template" module args.

levskaya committed 5 years ago

71aa41a6

Eval on ema instead of model parameters

marcvanzee committed 5 years ago

71ce0544

Merge pull request #415 from levskaya:linen3

a-googler committed 5 years ago

5ace793f

Add basic transform tests.

levskaya committed 5 years ago

7dd8d32c

Merge pull request #404 from wrzadkow:fix-optim-readme

a-googler committed 5 years ago

1f4b841a

internal change

levskaya committed 5 years ago

520a1cb8

marcvanzee committed 5 years ago

592fb12f

Re-enable checkpointing

marcvanzee committed 5 years ago

9423c786

Merge branch 'master' of https://github.com/google/flax into pixelcnn

marcvanzee committed 5 years ago

84d7f683

Fix argument name and its usage.

mohitreddy1996 committed 5 years ago

7ececa5f

Correct enum_values accepted - s/consine/cosine.

mohitreddy1996 committed 5 years ago

239e31cd

Refactor cifar10 to accept only FLAG values and move validations as part of FLAG definitions.

mohitreddy1996 committed 5 years ago

d883f3e8

Merge pull request #371 from google:linen2

a-googler committed 5 years ago

0132b3f2

Merge pull request #289 from mohitreddy1996:refactor-mnist

a-googler committed 5 years ago

cb3f3fa4

Merge branch 'master' into linen2

levskaya committed 5 years ago

b6bf5ee4

Remove the ModuleTemplate syntax for now.

levskaya committed 5 years ago

6bb417a2

Merge changes from jheek's functional-core branch.

levskaya committed 5 years ago

f8c70d67

Merge pull request #411 from jheek/linen2-template

levskaya committed 5 years ago

Verified 136a11fd

Merge pull request #412 from avital/xdist

avital committed 5 years ago

Verified 20a55357

Merge remote-tracking branch 'trunk/xdist' into refactor-mnist

avital committed 5 years ago

ac8ee8d5

Downgrade pytest-xdist. Will likely fix tests

avital committed 5 years ago

2f00d36a

Fix Polyak Averaging HOWTO

avital committed 5 years ago

12447487

Fix ensembling HOWTO

avital committed 5 years ago

b7732f9a

Newer Older