google/flax

Pull Requests Commits

a-googler committed 5 years ago

93ee347b

[tfds] Rename `tfds.features.text` -> `tfds.deprecated.text`

Conchylicultor committed 5 years ago

18958e24

revise check for JAX omnistaging

mattjj committed 5 years ago

c799b5d9

Add imagenet benchmark that uses fake data.

skye committed 5 years ago

7765b8ed

Merge pull request #376 from juliakreutzer:eos

a-googler committed 5 years ago

7c8a0acb

juliakreutzer committed 5 years ago

8766a4eb

consistent use of EOS id

juliakreutzer committed 5 years ago

86d4cad3

fix missing rng arg implementation in wmt example

levskaya committed 5 years ago

23a91dbc

Merge pull request #365 from tirkarthi:fix-warnings

a-googler committed 5 years ago

f8d3e11a

Merge pull request #363 from levskaya:v0.2

a-googler committed 5 years ago

79507ece

Fix syntax warning over comparison of literals using is.

tirkarthi committed 5 years ago

bfa58010

mark new version and add notes to changelog for v0.2

levskaya committed 5 years ago

3393e12e

Changed hardcoded type "float32" to dtype (models.py)

a-googler committed 5 years ago

a5dfa290

Remove expectations on secs_per_epoch metric.

a-googler committed 5 years ago

0f66ea70

Merge pull request #352 from wrzadkow:assert-lamb

a-googler committed 5 years ago

8ac99b8f

Merge pull request #358 from juliakreutzer:master

a-googler committed 5 years ago

b69f4038

Revert "Internal only change"

avital committed 5 years ago

5543fae9

Merge pull request #341 from DanGraur/fix-example-imgnet-training

avital committed 5 years ago

Verified 432d5ee2

extracted hard-coded hyperparameters into flags

juliakreutzer committed 5 years ago

3f33f80e

Start exporting event files to benchmark_output_dir as well, as this will help with easier integration with tensorboard.dev

a-googler committed 5 years ago

c4725328

Internal only change

a-googler committed 5 years ago

d3b6b359

Assert valid learning rate in LAMB optimizer

wrzadkow committed 5 years ago

5f7c2e26

Merge pull request #316 from danielsuo/dsuo/improve-check-howto-error

avital committed 5 years ago

Verified 8346a106

Update check_howto_diffs.sh

avital committed 5 years ago

Verified 4da1a18c

Update check_howto_diffs.sh

avital committed 5 years ago

Verified 09241205

Merge branch 'master' into dsuo/improve-check-howto-error

avital committed 5 years ago

Verified 31f8b37e

Merge pull request #313 from danielsuo/dsuo/howto-individual-pages

avital committed 5 years ago

Verified f4111921

Merge pull request #346 from DanGraur:jax-utils-assert-message-fix

a-googler committed 5 years ago

15223579

Start exporting parameters used by the benchmarks.

a-googler committed 5 years ago

060737d4

removed slashes from improved _prefetch assert.

levskaya committed 5 years ago

Verified 2e7012da

Older