google/flax

Pull Requests Commits

fix args for `create_learning_rate_schedule`

jaehlee committed 4 years ago

36a0bf59

match examples/wmt changes

jaehlee committed 4 years ago

97cd6657

Add marker file for PEP 561: Distributing and Packaging Type Information

manifest committed 4 years ago

Verified 7101225e

Merge pull request #1475 from jheek:fix-cloudpickle

a-googler committed 4 years ago

f558f49c

Updates README with new numbers/links.

andsteing committed 4 years ago

5c0b56e4

Update instructions for running lm1b example

jaehlee committed 4 years ago

27e640ea

update lm1b example to use Optax.

jaehlee committed 4 years ago

c086d0eb

Fixes typo in README.

andsteing committed 4 years ago

8a332b11

Updates wmt example to use Optax.

andsteing committed 4 years ago

4b1baddf

Fix cloudpickle serialization of wrapped Module methods.

jheek committed 4 years ago

b7571e44

Updates instructions to run on TPUv3-8 1VM.

andsteing committed 4 years ago

de929a18

Updates instructions for running wmt example.

andsteing committed 4 years ago

5bb89bb3

Bumps examples/cloud script versions.

andsteing committed 4 years ago

a6780b62

Merge pull request #1471 from elliotwaite:docs-fixes

a-googler committed 4 years ago

1a24c4d5

Removes jraph dependency

andsteing committed 4 years ago

Verified 1f3cf2bd

Merge pull request #1470 from ameya98/patch-1

andsteing committed 4 years ago

Verified 4e4a1f66

Merge branch 'main' of https://github.com/google/flax into docs-fixes

elliotwaite committed 4 years ago

b532c97e

Several small docs fixes

elliotwaite committed 4 years ago

d9e59d81

Add jraph to test dependencies.

ameya98 committed 4 years ago

Verified 1f50ddc6

ZYHowell committed 4 years ago

721e6875

Merge pull request #1466 from cheshire:patch-1

a-googler committed 4 years ago

ccd36132

Update imagenet_fake_data_benchmark to include warmup phase

cheshire committed 4 years ago

Verified 08a6d50d

add enable_kwargs

ZYHowell committed 4 years ago

39492d5e

Add some type annotations to flax.serialization

prazek committed 4 years ago

c63d9c69

Chunk and unchunk msgpack_serialize and msgpack_restore.

prazek committed 4 years ago

0f386255

Improves type annotations and `safe_normpath()` in `flax.training.checkpoints`.

andsteing committed 4 years ago

9871df7f

Merge pull request #1458 from slowy07:minor-fixing

a-googler committed 4 years ago

8b2e56e9

fix: typo grammar commented code

slowy07 committed 4 years ago

c5ddf64b

kwargs in remat and pack

ZYHowell committed 4 years ago

3cea8ad2

Make save_checkpoint more robust to non normal paths.

a-googler committed 4 years ago

095517e6

Newer Older