google/flax

Pull Requests Commits

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

Merge pull request #1497 from INF800:initializers/add-all-from-jax

a-googler committed 4 years ago

1a117d0a

Merge pull request #1511 from jheek:fix-embedding-hash

a-googler committed 4 years ago

8346ef03

Merge pull request #1499 from levskaya:rematfix

a-googler committed 4 years ago

be522b51

remove unnecessary character in README.md

jaehlee committed 4 years ago

e01e2bcb

Update flax/examples parsing JAX config flags

jaehlee committed 4 years ago

8cf4eab6

Plumb through the prevent_cse kwarg for jax remat.

levskaya committed 4 years ago

7ed82a49

Newer Older