google/flax

Pull Requests Commits

Move latest_checkpoint functionality to its own method, similar to tf.train.latest_checkpoint.

cghawthorne committed 5 years ago

c289bf11

Merge pull request #872 from google:marcvanzee-patch-2

a-googler committed 5 years ago

435a188f

Merge pull request #871 from google:marcvanzee-patch-1

a-googler committed 5 years ago

e533f664

Update comment on sync_batch_stats

marcvanzee committed 5 years ago

f4d234b6

Adds profiler.collect() and makes Profile(logdir) mandatory.

andsteing committed 5 years ago

7cdfcafe

Merge pull request #937 from levskaya:transform_improvements2

a-googler committed 5 years ago

38323b2b

Generalize transform handling for nested attribute submodules.

levskaya committed 5 years ago

cf2a2893

Merge pull request #931 from google:marcvanzee-patch-5

a-googler committed 5 years ago

8f9f7ece

Adds comment to "focus" of OptimizerDef.create()

marcvanzee committed 5 years ago

Verified 5b90b7d1

Merge pull request #925 from avital:just-sphinx-2

a-googler committed 5 years ago

30b21fa5

Use doctest for new HOWTOs, instead of our old diff system

avital committed 5 years ago

265afe1f

add automatic flattening of ConfigDict-like structures for hparam logging.

cghawthorne committed 5 years ago

3e36db3e

Clarifies usage of using jax.jit in Module.init()

marcvanzee committed 5 years ago

ae74bb3a

Merge pull request #841 from SamuelMarks:args-for-google-style-docstrings

a-googler committed 5 years ago

e2cb2844

Merge pull request #909 from google:multi-optim

a-googler committed 5 years ago

2c73c2ca

marcvanzee committed 5 years ago

508434b4

Updates optimizer docstrings and RTD

marcvanzee committed 5 years ago

bd285fc1

Update comment on Models

marcvanzee committed 5 years ago

Verified 6a7c5f0a

Add dtype option for momentum buffers to adafactor.

akolesnikoff committed 5 years ago

e14840ea

Merge pull request #884 from avital/remove-examples-take-2

avital committed 5 years ago

Verified bb25a483

Rename linen_examples -> examples part 2:

avital committed 5 years ago

4d593fe7

Copybara import of the project:

avital committed 5 years ago

d068512a

Improve documentation and minor pystyle violations.

a-googler committed 5 years ago

cf16fa26

Merge pull request #829 from andsteing:mnist

a-googler committed 5 years ago

c84fa7d7

Add comment to prefetch_to_device explaining it is mostly useful for GPUs (not TPUs).

marcvanzee committed 5 years ago

Verified b4d1895d

[README.md] Add missing word and adjust surround word wrap

SamuelMarks committed 5 years ago

Verified 4ba70974

Merge pull request #838 from google:marcvanzee-patch-1

a-googler committed 5 years ago

07b29805

Merge branch 'master' of https://github.com/google/flax into marcvanzee-patch-1

marcvanzee committed 5 years ago

3752ea04

Merge pull request #822 from jheek:linen-ModelParamTraversal

a-googler committed 5 years ago

87276132

[*.py] Rename "Arguments:" to "Args:"

SamuelMarks committed 5 years ago

Verified 2f352057

Newer Older