google/flax

Pull Requests Commits

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

Fixes some typos mentioned in #828.

andsteing committed 5 years ago

b1e6d4f2

Shows table of content by default.

andsteing committed 5 years ago

afa513df

update CHANGELOG

jheek committed 5 years ago

5a7d0fa5

ModelParamTraversal for linen

jheek committed 5 years ago

87d91567

Merge pull request #840 from jheek:build-status-badge

a-googler committed 5 years ago

cb128e6b

Update README.md

avital committed 5 years ago

Verified 8aa7e319

build status badge

jheek committed 5 years ago

3272723f

Merge pull request #839 from jheek:fix-normalization-test-2

a-googler committed 5 years ago

27e94552

Fix another case of isInstance test

jheek committed 5 years ago

071d1051

Merge pull request #836 from jheek:fix-normalization-test

a-googler committed 5 years ago

f3c05408

Add empty lines to CHANGELOG

marcvanzee committed 5 years ago

Verified 30598263

Newer Older