google/flax

Pull Requests Commits

Internal changes linen_examples/mnist.

a-googler committed 5 years ago

56c223d0

Merge pull request #463 from mohitreddy1996:mnist-mlcollections

a-googler committed 5 years ago

7ee14bc3

Merge pull request #472 from andsteing:master

a-googler committed 5 years ago

6ab7b1ab

Merge pull request #475 from wrzadkow:add-__version__

a-googler committed 5 years ago

dc52011e

Bugfix mistake previous rebase.

andsteing committed 5 years ago

01376b3f

Fixed HOWTOs even more.

andsteing committed 5 years ago

2d83fc26

andsteing committed 5 years ago

10d609b3

Removes tf.enable_v2_behavior().

andsteing committed 5 years ago

558a9aa5

Hide GPUs from TensorFlow and use default v2.

andsteing committed 5 years ago

b5ca503e

Expose __version__ attribute

wrzadkow committed 5 years ago

05b3ee45

Merge pull request #474 from jheek:fix-scan-broadcast

a-googler committed 5 years ago

d38659c6

Merge pull request #287 from mohitreddy1996:refactor-imagenet

a-googler committed 5 years ago

c16e5937

Fix scan broadcast issue

jheek committed 5 years ago

0cd39d05

mohitreddy1996 committed 5 years ago

44ba2a51

Resolve merge conflicts.

mohitreddy1996 committed 5 years ago

aa06c7a1

Fix initialization RNGs to work with omnistaging for jitted inits.

levskaya committed 5 years ago

139d02fd

Merge pull request #468 from marcvanzee/master

marcvanzee committed 5 years ago

Verified 3c929273

marcvanzee committed 5 years ago

9bc94983

marcvanzee committed 5 years ago

efbf5782

marcvanzee committed 5 years ago

0131c627

Merge pull request #471 from google/fix-docs

avital committed 5 years ago

Verified f47f8b79

force upgrade pygments

jheek committed 5 years ago

a98d85d6

marcvanzee committed 5 years ago

b8a96e0f

Fix sampling howto

marcvanzee committed 5 years ago

1e4036f5

Merge pull request #469 from jheek:linen-docs-udpate

a-googler committed 5 years ago

cd1d5313

update linen intro link

jheek committed 5 years ago

b543effe

Add linen intro notebook

jheek committed 5 years ago

045a0f96

marcvanzee committed 5 years ago

fa73a2ad

Remove some more code

marcvanzee committed 5 years ago

b7ce5b84

Remove unnecessary shared()

marcvanzee committed 5 years ago

2c27212e

Older