google/flax

Pull Requests Commits

Merge pull request #446 from j-towns:minor-pcnn-fixes

a-googler committed 5 years ago

c7d6559d

Merge pull request #441 from levskaya:linen_wmt

a-googler committed 5 years ago

c3afa829

Minor style tweaks in pixelcnn

j-towns committed 5 years ago

5990cc64

using activation.sigmoid

mbz committed 5 years ago

Verified ed4dcaaf

Using nn.sigmoid instead of jnn.sigmoid

mbz committed 5 years ago

Verified b65329a0

Add single input conv and conv_transpose support

vballoli committed 5 years ago

aa3f030e

Merge pull request #343 from lucasb-eyer:patch-4

a-googler committed 5 years ago

89b07cb2

Merge branch 'master' into patch-4

jheek committed 5 years ago

Verified 4b40e549

Merge pull request #433 from jheek:scope-aliasing

a-googler committed 5 years ago

99113ceb

Update attention layers with simpler flat forms, update wmt example.

levskaya committed 5 years ago

c312382d

jheek committed 5 years ago

2764f710

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

marcvanzee committed 5 years ago

aaced05b

Improve error message

marcvanzee committed 5 years ago

2f39f535

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

fix ordering issue

jheek committed 5 years ago

2d00ef68

Adds MNIST to Linen examples

marcvanzee committed 5 years ago

b5a7d3de

fix remaining TODOs

jheek committed 5 years ago

d0a888ae

Newer Older