google/flax

Pull Requests Commits

Avoid passing python collections to jax reductions

a-googler committed 5 years ago

05a2add6

Merge pull request #435 from marcvanzee:vae_linen

a-googler committed 5 years ago

aee7fbe9

Merge pull request #435 from marcvanzee:vae_linen

a-googler committed 5 years ago

d45c21d4

Recover accidentally commented out jit

marcvanzee committed 5 years ago

79829fd7

Adds Linen VAE to linen examples

marcvanzee committed 5 years ago

1bcefdab

Merge pull request #416 from marcvanzee:pixelcnn

a-googler committed 5 years ago

00f5faa1

Adds requirements.txt

marcvanzee committed 5 years ago

a2dd38c0

marcvanzee committed 5 years ago

b4ec2c58

Fix sampler and add newly sample sample.png

marcvanzee committed 5 years ago

1dbdbad3

Merge pull request #434 from marcvanzee:mnist_linen

a-googler committed 5 years ago

24617838

Remove accidental changes to original pixelcnn

marcvanzee committed 5 years ago

95ae7acd

Rename params to variables in tests

marcvanzee committed 5 years ago

abe14526

marcvanzee committed 5 years ago

d68e26df

marcvanzee committed 5 years ago

67eb8e14

Adds MNIST to Linen examples

marcvanzee committed 5 years ago

b5a7d3de

Incorporate comments

marcvanzee committed 5 years ago

103dbcd9

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

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

Older