google/flax

Pull Requests Commits

Merge pull request #436 from marcvanzee:errormsg

a-googler committed 5 years ago

efc4ba74

named_call uses layer name rather than layer class name.

levskaya committed 5 years ago

efe9d0a6

Fix module test after change in error message

marcvanzee committed 5 years ago

0f5dcefa

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

marcvanzee committed 5 years ago

bac79a0b

Renames get_param to get_initial_params in seq2seq and mnist Linen examples

marcvanzee committed 5 years ago

c33df578

fix import bug and enable profiling via environment variable.

levskaya committed 5 years ago

0fb2b89d

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

Fix comments: tensorflow.compat.v2 -> tensorflow.

mohitreddy1996 committed 5 years ago

72ec6a38

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

Newer Older