google/flax

Pull Requests Commits

Adding the tpu vm instructions on imagenet.

sshahrokhi committed 5 years ago

82e97982

Merge pull request #1328 from n2cholas:docstrings

a-googler committed 5 years ago

6fb839c6

Fix minor sow example issues.

n2cholas committed 5 years ago

8a18e8e8

Update sow example to account for immutable intermediates during init.

n2cholas committed 5 years ago

c3556633

Make Module bind and sow examples runnable.

n2cholas committed 5 years ago

f430453a

Make README examples runnable.

n2cholas committed 5 years ago

49380ac6

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

jheek committed 5 years ago

Verified 56c83723

Remove enable_omnistaging

jheek committed 5 years ago

607747d6

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

jheek committed 5 years ago

Verified 6ba7d6dc

jheek committed 5 years ago

38297d87

Merge pull request #1332 from hmph:patch-2

a-googler committed 5 years ago

36fccf2c

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

a-googler committed 5 years ago

3d639e80

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

a-googler committed 5 years ago

22f41b6f

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

marcvanzee committed 5 years ago

Verified a40a3c39

8bitmp3 committed 5 years ago

Verified 50032c74

Capitalize JVP-VJPs

8bitmp3 committed 5 years ago

Verified 67fd2c9b

Enhance JAX for the Impatient doc (Flax)

8bitmp3 committed 5 years ago

Verified 6816a0ff

Updated the citation version to the most receent flax version

mplemay committed 5 years ago

01fbad59

Fix potential race condition in restoring checkpoints.

tutmann committed 5 years ago

572f5181

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

jheek committed 5 years ago

Verified e56340f2

Add atari rom depencency

jheek committed 5 years ago

50ce7a01

Merge pull request #1276 from andsteing:clu

a-googler committed 5 years ago

4b40bf4c

Merge remote-tracking branch 'upstream/master' into clu

andsteing committed 5 years ago

503fe76f

[flax] Add messages to some assertions in `pool` to help with debugging.

a-googler committed 5 years ago

1dc3d541

Merge pull request #1324 from n2cholas:merge_param_import

a-googler committed 5 years ago

276ee008

Correct a small mistake in error documentation

anuragarnab committed 5 years ago

Verified fc063079

Merge pull request #1267 from jheek/lift-md

jheek committed 5 years ago

Verified d545b0ee

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

andsteing committed 5 years ago

Verified 6111a4b2

read copybara config from current CL for dry-run presubmit

marcvanzee committed 5 years ago

43987f8d

Expose merge_param at top level of flax.linen.

n2cholas committed 5 years ago

bf1c2df8

Newer Older