google/flax

Pull Requests Commits

Removed a superfluous assignment in the examples/imagenet/train.py source.

Dan Graur committed 6 years ago

4c2319fd

jheek committed 6 years ago

ee20ca22

use staging jaxpr trace in lazy init

jheek committed 6 years ago

6ccd6a60

fix omnistaging issues

jheek committed 6 years ago

1f553886

Align layernorm dtype handling with batchnorm (i.e., use requested dtype for layernorm outputs, even though intermediate computations are f32).

jekbradbury committed 6 years ago

ed42d067

Updating OSS workflow for docs

a-googler committed 6 years ago

1c509ed4

Merge pull request #328 from lucasb-eyer/patch-3

MostafaDehghani committed 6 years ago

Verified 4578df43

Also add link to Jax's dot_general doc.

lucasb-eyer committed 6 years ago

Verified 23b3be96

Add link to DotGeneral in DenseGeneral doc.

lucasb-eyer committed 6 years ago

Verified 78945d4c

Flax README typos cleanup.

dbieber committed 6 years ago

fbdf8513

Merge pull request #301 from ayush-1506:batchnorm_error_msg

a-googler committed 6 years ago

484cc116

Merge pull request #315 from google:mohitreddy1996-patch-1

a-googler committed 6 years ago

639821e1

Concat the distributed batchnorm to make it more efficient.

a-googler committed 6 years ago

a3a6f7ba

Export model_name from the MNIST benchmark.

a-googler committed 6 years ago

59e480f7

Export model_name from the MNIST benchmark

a-googler committed 6 years ago

81752d83

Remove loss scaling from README

mohitreddy1996 committed 6 years ago

Verified 95844ce2

Make clear that we are undoing git apply if tests fail in check_howto_diffs.sh

danielsuo committed 6 years ago

b8610d70

Improve error message in check_howto_diffs.sh

danielsuo committed 6 years ago

7c649dd1

Remove loss_scaling flag from imagenet benchmark

mohitreddy1996 committed 6 years ago

Verified 5a1787b3

Move howto index page

danielsuo committed 6 years ago

f62618c1

Add new howto docs

danielsuo committed 6 years ago

7d5fdc36

Move howto docs to their own directory

danielsuo committed 6 years ago

2774dcbf

Add check for howto documentation file

danielsuo committed 6 years ago

21c396ef

Add checkingpointing page

danielsuo committed 6 years ago

b6cacae4

Give each howto its own page

danielsuo committed 6 years ago

b55e1864

Merge pull request #304 from google/test_314238108

levskaya committed 6 years ago

Verified 5493b917

fix create vs fill race condition for multihost wmt training.

levskaya committed 6 years ago

a4b095f3

Add more meaningful error message to batchnorm

ashridh committed 6 years ago

6cbc6331

Merge pull request #285 from kshithijiyer/patch-1

avital committed 6 years ago

Verified 8901bc07

Merge pull request #297 from danielsuo/dsuo/gitignore-rej

avital committed 6 years ago

Verified 967103ee

Newer Older