google/flax

Pull Requests Commits

Rename host_count and host_id to process_count and process_index.

dbieber committed 4 years ago

2ec5fbff

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

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

andsteing committed 4 years ago

503fe76f

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

a-googler committed 4 years ago

1dc3d541

Merge pull request #1324 from n2cholas:merge_param_import

a-googler committed 4 years ago

276ee008

Merge pull request #1267 from jheek/lift-md

jheek committed 4 years ago

Verified d545b0ee

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

andsteing committed 4 years ago

Verified 6111a4b2

read copybara config from current CL for dry-run presubmit

marcvanzee committed 4 years ago

43987f8d

Expose merge_param at top level of flax.linen.

n2cholas committed 4 years ago

bf1c2df8

Merge pull request #1312 from JamesAllingham/add-bn-vmap-to-state-howto

jheek committed 4 years ago

Verified c53c1c53

Merge pull request #1314 from ilopezgp/docs_spelling_fix

jheek committed 4 years ago

Verified d5f12e92

Merge pull request #1318 from google/new-version

marcvanzee committed 4 years ago

Verified eba1dbaa

Update CHANGELOG.md

marcvanzee committed 4 years ago

Verified 3f2775ee

Update CHANGELOG.md

marcvanzee committed 4 years ago

Verified 13f16c18

Apply suggestions from code review

marcvanzee committed 4 years ago

Verified e54f8e70

Update CHANGELOG.md

marcvanzee committed 4 years ago

Verified 7788281d

Updates references to flax.training.train_state.

andsteing committed 4 years ago

b27e7846

Renames `train_state` to `state`.

andsteing committed 4 years ago

ea47d81c

Merge pull request #1327 from google/marcvanzee-patch-10

marcvanzee committed 4 years ago

Verified 0f9cc8b3

Don't use newest TF (see #1326)

marcvanzee committed 4 years ago

Verified 36616e11

Merge pull request #1311 from tirkarthi:fix-threading

a-googler committed 4 years ago

491ce187

Flax version 0.3.4

marcvanzee committed 4 years ago

c4222e01

Merge pull request #1302 from andsteing:mnist

a-googler committed 4 years ago

4ae9143f

Minor spelling corrections, JAX for the impatient docs

ilopezgp committed 4 years ago

5511ac33

Merge pull request #1307 from jheek:typo-fixes

a-googler committed 4 years ago

9da95cdd

add more info on PRNG splitting

jheek committed 4 years ago

c07dceb8

Add DenyList example

jheek committed 4 years ago

52415bd5

Older