google/flax

Pull Requests Commits

Adds a link to imagenet Colab showing use of TPUs.

andsteing committed 4 years ago

47af76a9

Merge pull request #1117 from marcvanzee:param-docstring

a-googler committed 4 years ago

b1c02160

Merge pull request #1121 from brettkoonce:lm1b-readme-tweak

a-googler committed 4 years ago

f90f0344

marcvanzee committed 4 years ago

1691766b

Merge remote-tracking branch 'upstream/master' into module-errors

marcvanzee committed 4 years ago

609b3bdc

Implement comments on PR

marcvanzee committed 4 years ago

a56d22c5

Merge pull request #1120 from jheek:fix-checkpoint-overwrite

a-googler committed 4 years ago

dcacf694

Merge pull request #1127 from marcvanzee:fix-error

a-googler committed 4 years ago

19967e52

restore batch size overrider

brettkoonce committed 4 years ago

8848a527

Merge pull request #1116 from jheek:remove-FrozenDict-mutability-checks

a-googler committed 4 years ago

0ef44eb4

Remove FrozenDict checks to determine mutability.

jheek committed 4 years ago

1fc88713

Adds is_initialized flag to module.py

marcvanzee committed 4 years ago

d5c6c50f

marcvanzee committed 4 years ago

703401e4

Merge pull request #1119 from avital:remove-deprecated-examples

a-googler committed 4 years ago

d047b823

Copybara import of the project:

andsteing committed 4 years ago

03d2b983

lm1b: tweak readme command

brettkoonce committed 4 years ago

82aa98eb

support batch dimension in image summaries

iansimon committed 4 years ago

6e0c0f46

Merge pull request #1112 from brettkoonce:wmt-tweak

a-googler committed 4 years ago

ae1d860b

save_checkpoint behavior was ill-defined when later checkpoint already exist

jheek committed 4 years ago

44a03d83

Remove deprecated LM1b example

avital committed 4 years ago

b02a2062

Remove legacy HOWTOs

avital committed 4 years ago

f44752e3

Update docstrings on module.py

marcvanzee committed 4 years ago

825fc2a0

Merge pull request #1113 from brettkoonce:mnist-readme-tweak

a-googler committed 4 years ago

15b6229d

marcvanzee committed 4 years ago

f04659c7

Add pathlib / fspath compatibility to SummaryWriter

Conchylicultor committed 4 years ago

afa4a2cd

Use NameInUseError for params,vars,submodules

marcvanzee committed 4 years ago

cc063f02

mnist: tweak demo call

brettkoonce committed 4 years ago

4f0760cc

wmt: tweak readme example config

brettkoonce committed 4 years ago

91a937cb

Add documentation for transforms

jheek committed 4 years ago

1cd3e637

Merge pull request #1100 from jheek:improve-bind-docs

a-googler committed 4 years ago

748f7a38

Newer Older