google/flax

Pull Requests Commits

Incorporate comments

marcvanzee committed 5 years ago

103dbcd9

Update recurrent.py

mbz committed 5 years ago

Verified cd588e53

renaming to ConvLSTM

mbz committed 5 years ago

Verified 3c24ff13

adding ConvLSTM to nn

mbz committed 5 years ago

Verified 30f73cac

adding copyright back!

mbz committed 5 years ago

Verified b4c2cb3f

Update nn_test.py

mbz committed 5 years ago

Verified c675b936

Merge pull request #414 from mohitreddy1996:cifar-unittesting

a-googler committed 5 years ago

294575ac

Merge pull request #418 from mohitreddy1996:sst2-unittesting

a-googler committed 5 years ago

61db61ae

Add the Adafactor optimizer.

levskaya committed 5 years ago

357117c8

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

marcvanzee committed 5 years ago

6df99e44

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

marcvanzee committed 5 years ago

827abd8a

Merge pull request #429 from google/levskaya-patch-1

avital committed 5 years ago

Verified 7be4c234

Merge pull request #425 from jheek:harmonized-scan

a-googler committed 5 years ago

9b846321

add scope aliasing to lift.pack

jheek committed 5 years ago

143626bb

Update run_all_tests.sh to run linen_examples tests.

levskaya committed 5 years ago

Verified b1dd3fe3

Add full ImageNet benchmark, run every 24h.

avital committed 5 years ago

fe04d0e3

Incorporate PR review comments

marcvanzee committed 5 years ago

0c96dafa

fix seq2seq example

jheek committed 5 years ago

2d0bdb62

jheek committed 5 years ago

3443c466

harmonized scan

jheek committed 5 years ago

d1e6c2a4

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

marcvanzee committed 5 years ago

8190f623

Merge pull request #417 from levskaya:linen3

a-googler committed 5 years ago

14883b01

Adding Conv2DLSTM

mbz committed 5 years ago

Verified 0bc9cd44

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

Newer Older