google/flax

Pull Requests Commits

Create new bug template "feature request"

marcvanzee committed 5 years ago

Verified db881140

Add implementation as linen for imagenet benchmark. Also add epochs=2 as parameters.

Mohit Reddy committed 5 years ago

a1090ef5

a-googler committed 5 years ago

8debea1f

Merge pull request #516 from jheek/fix-warning

jheek committed 5 years ago

Verified 5f70a290

Fix warning about float64 types

jheek committed 5 years ago

071d5385

Merge pull request #514 from mohitreddy1996:hparams-examples

a-googler committed 5 years ago

43d78f06

Write hparams MNIST and ImageNet linen examples.

mohitreddy1996 committed 5 years ago

006a9f3e

Updates link to CIFAR example.

andsteing committed 5 years ago

5c1b878e

Merge pull request #496 from andsteing:examples

a-googler committed 5 years ago

5e7eeb77

Merge pull request #507 from andsteing:doc

a-googler committed 5 years ago

aa4d16cc

Adds link to dev/README.md

andsteing committed 5 years ago

8ef59910

Explicitly mentions run_all_test.sh script.

andsteing committed 5 years ago

Verified ac0ba156

Respect original setup for "no tests found" errors.

andsteing committed 5 years ago

44a2cfba

Fix typo README.md

andsteing committed 5 years ago

Verified 0c6be7fe

Updates README "Getting involved".

andsteing committed 5 years ago

75cef089

Merge pull request #454 from marcvanzee:automasking

a-googler committed 5 years ago

3939192f

Change automasking shape and use list instead of np.array

marcvanzee committed 5 years ago

bee1d654

Fix tests for automasking

marcvanzee committed 5 years ago

c4e36ad0

Fix automasking

marcvanzee committed 5 years ago

a0774a26

Merge pull request #505 from marcvanzee/pipy

marcvanzee committed 5 years ago

Verified 4f5714f7

Update CHANGELOG v0.2.2

marcvanzee committed 5 years ago

7427b8f8

Update CHANGELOG v0.2.2

marcvanzee committed 5 years ago

d2c5f492

Merge pull request #470 from wrzadkow:rl-example-ppo

a-googler committed 5 years ago

45937af8

Specify static_argnums with proper int

wrzadkow committed 5 years ago

f3a9d03e

Compile vectorized code instead of vectorizing compiled code

wrzadkow committed 5 years ago

d444075e

Test: use assertEqual and clip rewards when testing them

wrzadkow committed 5 years ago

d2eae5c0

Update CHANGELOG v0.2

marcvanzee committed 5 years ago

e00e2494

Merge pull request #470 from wrzadkow:rl-example-ppo

a-googler committed 5 years ago

fed1aafd

Fix automasking: hidden size was too small

marcvanzee committed 5 years ago

22146233

Refactor a long statement

wrzadkow committed 5 years ago

315902bd

Older