google/flax

Pull Requests Commits

Update index.rst

andsteing committed 4 years ago

Verified baaf2de8

Replaces link to README with link to Github repo

andsteing committed 4 years ago

Verified de72f284

Merge pull request #1354 from jheek:fix-incorrect-named-tuple-conversion

a-googler committed 4 years ago

88464610

Merge pull request #1351 from lucasb-eyer:patch-7

a-googler committed 4 years ago

4548bbdb

Fix NamedTuples being converted to tuples in Module

jheek committed 4 years ago

6759522a

Merge pull request #1350 from google/andsteing-patch-1

jheek committed 4 years ago

Verified 5ebf14f2

lucasb-eyer committed 4 years ago

Verified 25804a14

Adds reference to matthias-wright/flaxmodels

andsteing committed 4 years ago

Verified 7392f1e4

Adding the tpu vm instructions on imagenet.

sshahrokhi committed 4 years ago

82e97982

Merge pull request #1328 from n2cholas:docstrings

a-googler committed 4 years ago

6fb839c6

Fix minor sow example issues.

n2cholas committed 4 years ago

8a18e8e8

Update sow example to account for immutable intermediates during init.

n2cholas committed 4 years ago

c3556633

Make Module bind and sow examples runnable.

n2cholas committed 4 years ago

f430453a

Make README examples runnable.

n2cholas committed 4 years ago

49380ac6

Merge pull request #1342 from jheek/remove-explicit-omnistaging

jheek committed 4 years ago

Verified 56c83723

Remove enable_omnistaging

jheek committed 4 years ago

607747d6

Merge pull request #1343 from jheek/fix-atari-py-dep

jheek committed 4 years ago

Verified 6ba7d6dc

jheek committed 4 years ago

38297d87

Merge pull request #1332 from hmph:patch-2

a-googler committed 4 years ago

36fccf2c

Merge pull request #1329 from google:marcvanzee-patch-10

a-googler committed 4 years ago

3d639e80

Merge pull request #1337 from mplemay:updated-flax-readme-version

a-googler committed 4 years ago

22f41b6f

Merge pull request #1338 from 8bitmp3/patch-1

marcvanzee committed 4 years ago

Verified a40a3c39

8bitmp3 committed 4 years ago

Verified 50032c74

Capitalize JVP-VJPs

8bitmp3 committed 4 years ago

Verified 67fd2c9b

Enhance JAX for the Impatient doc (Flax)

8bitmp3 committed 4 years ago

Verified 6816a0ff

Updated the citation version to the most receent flax version

mplemay committed 4 years ago

01fbad59

Fix potential race condition in restoring checkpoints.

tutmann committed 4 years ago

572f5181

Merge pull request #1336 from jheek/fix-ppo-

jheek committed 4 years ago

Verified e56340f2

Add atari rom depencency

jheek committed 4 years ago

50ce7a01

Merge pull request #1276 from andsteing:clu

a-googler committed 4 years ago

4b40bf4c

Older