google/flax

Pull Requests Commits

Make sure itemization appears in RTD correctly

marcvanzee committed 4 years ago

Verified 5fa8b29f

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

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

Update CHANGELOG.md

marcvanzee committed 4 years ago

Verified 074caaed

Add vmap + BatchNorm example to the state howto

JamesAllingham committed 4 years ago

00e9fea7

Use notify_all instead of notifyAll method that was deprecated in Python 3.10.

tirkarthi committed 4 years ago

c1e5bf62

Merge pull request #1306 from bastings:sst2_optax

a-googler committed 4 years ago

48b5707a

Merge pull request #1291 from avital:attention-weights

a-googler committed 4 years ago

e95acab1

Add more assertions to dot_product_attention_weights

avital committed 4 years ago

710d2023

bastings committed 4 years ago

a64d88d5

Move to use Optax optimizer.

bastings committed 4 years ago

afd14928

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

bastings committed 4 years ago

cf27baf3

Correct assertion string

avital committed 4 years ago

Verified 43a98165

jheek committed 4 years ago

1a0a0136

Renames jax.{host_id,host_count} -> jax.{process_index,process_count} & bumps jaxlib version.

andsteing committed 4 years ago

d969e64d

Fixes type examples/mnist/README.md

andsteing committed 4 years ago

Verified ecf4c592

Switch to Optax optimizer and use TrainState.

bastings committed 4 years ago

a3990ed1

Merge pull request #1295 from n2cholas:scan-docs

a-googler committed 4 years ago

0082b90c

Older