google/flax

Pull Requests Commits

Fix seq2seq implicit rank promotion

jheek committed 4 years ago

ebe84238

Remove Implicit rank promotion

jheek committed 4 years ago

a401e824

Merge pull request #1531 from jheek:fix-lifting-rng-splits

a-googler committed 4 years ago

cf638d18

Remove users of jax.api.* symbols, in preparation for removing the deprecated jax.api name.

hawkinsp committed 4 years ago

e0a618a9

Remove unnecessary splitting of rngs in lifting logic.

jheek committed 4 years ago

0d69f1a2

Dynamically check whether named call is enabled.

jheek committed 4 years ago

6c46a167

Fix temperature sampler loop termination condition:

renjie-liu committed 4 years ago

4ccd5fff

GPU Mixed precision training for WMT

jheek committed 4 years ago

777ee543

Add optional keep_every_n_steps argument to checkpointing. When enabled, save_checkpoint will keep checkpoints every n steps.

tutmann committed 4 years ago

f92f936b

Merge pull request #1525 from levskaya:modelparamtraversal

a-googler committed 4 years ago

4d7d4580

Merge pull request #1526 from levskaya/pytestinifix

levskaya committed 4 years ago

Verified 37cc870e

Clean up pytest.ini

levskaya committed 4 years ago

965c4fe6

Separate ModelParamTraversal into traverse_util.

levskaya committed 4 years ago

31a1cf8f

Merge pull request #1524 from levskaya/logosvg

levskaya committed 4 years ago

Verified f7528664

Add flax logo in SVG and 3 PNG sizes.

levskaya committed 4 years ago

77e0f6ef

Merge pull request #1521 from google/levskaya-patch-1

marcvanzee committed 4 years ago

Verified 6d0a8079

Merge pull request #1519 from levskaya:nowrap

a-googler committed 4 years ago

aefe735c

Add nowrap decorator for module methods.

levskaya committed 4 years ago

fa348ba1

Merge pull request #1522 from google:remove_host_id

a-googler committed 4 years ago

a014c3b4

Merge pull request #1457 from ZYHowell:pack_kwargs

a-googler committed 4 years ago

a405dd17

remove remaining uses of deprecated host_id

levskaya committed 4 years ago

6775cab5

Correct incorrect shape note in ensembling.rst

levskaya committed 4 years ago

Verified d67ed87c

Merge pull request #1518 from romanngg:patch-2

a-googler committed 4 years ago

69d60a8c

Merge pull request #1514 from jaehlee:main

a-googler committed 4 years ago

65fa1395

Pad with zeros matching image type

romanngg committed 4 years ago

Verified a3a0a682

ZYHowell committed 4 years ago

Verified d9b27392

Merge pull request #1516 from romanngg:patch-1

a-googler committed 4 years ago

02956d81

Move import numpy to top

romanngg committed 4 years ago

Verified a30794bd

Merge branch 'main' into pack_kwargs

ZYHowell committed 4 years ago

Verified f56e52b6

Do not call `jnp` on shapes in AutoEncoder example

romanngg committed 4 years ago

Verified ca72d0a9

Newer Older