google/flax

Pull Requests Commits

Reduce the accuracy target for mnist, as it is 0.2576 on TPU.

a-googler committed 5 years ago

8c7bc1f3

Benchmarks class should scrap only the TF summary files and only scalar values from them.

a-googler committed 5 years ago

e5dc163d

[tfds] Fix tfds.as_numpy which now returns a reusable iterable.

Conchylicultor committed 5 years ago

e25e20c8

Improve partial_eval_by_shape compatibility with https://github.com/google/jax/pull/3370 and custom derivative rules.

mattjj committed 5 years ago

46e9c0b2

non bottleneck block for ImageNet example

jheek committed 5 years ago

ef7dcb0e

Improve partial_eval_by_shape compatibility with https://github.com/google/jax/pull/3370 and custom derivative rules.

mattjj committed 5 years ago

472b82a2

Merge pull request #362 from levskaya:poolingfix

a-googler committed 5 years ago

10491c2e

Merge pull request #401 from avital:fix-docs

a-googler committed 5 years ago

d505aa31

Merge pull request #400 from google/yapf

avital committed 5 years ago

Verified 0330f884

Bring back docs changes that were accidentally reverted

avital committed 5 years ago

4d0043ca

Add .style.yapf config that matches the Google internal style guide.

Marvin182 committed 5 years ago

2d04bef8

Merge pull request #394 from google:cifar-sota-repo

a-googler committed 5 years ago

277877cd

fix pooling ops with explicit padding

levskaya committed 5 years ago

18a9ae52

Fix in serialize and deserialize collections.

shivaniag committed 5 years ago

6b58fc4e

Link to new CIFAR SoTA repo by @PForet

avital committed 5 years ago

Verified b0f5278d

Merge pull request #383 from andsteing/master

avital committed 5 years ago

Verified b125c062

Merge pull request #337 from google/avital-link-to-discussions

avital committed 5 years ago

Verified d65be34b

Merge pull request #384 from Marvin182/master

Marvin182 committed 5 years ago

Verified e7f1754a

fix partial eval for custom_jvp and omnistaging changes

levskaya committed 5 years ago

44b07d02

[tfds] Rename `tfds.features.text` -> `tfds.deprecated.text`

Conchylicultor committed 5 years ago

18958e24

Add .style.yapf config that matches the Google internal style guide.

Marvin182 committed 5 years ago

7bcd502f

Fix typo wmt/README.md link

andsteing committed 5 years ago

48464d57

Add overview to examples/README.md

andsteing committed 5 years ago

41342608

revise check for JAX omnistaging

mattjj committed 5 years ago

c799b5d9

Add imagenet benchmark that uses fake data.

skye committed 5 years ago

7765b8ed

Merge pull request #376 from juliakreutzer:eos

a-googler committed 5 years ago

7c8a0acb

juliakreutzer committed 5 years ago

8766a4eb

consistent use of EOS id

juliakreutzer committed 5 years ago

86d4cad3

fix missing rng arg implementation in wmt example

levskaya committed 5 years ago

23a91dbc

Merge pull request #365 from tirkarthi:fix-warnings

a-googler committed 5 years ago

f8d3e11a

Older