google/flax

Pull Requests Commits

Split the attention softmax so that the expensive elementwise division happens on an array that's O(N) rather than O(N^2)

jekbradbury committed 5 years ago

1b2754a9

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

jekbradbury committed 5 years ago

ed42d067

Updating OSS workflow for docs

a-googler committed 5 years ago

1c509ed4

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

MostafaDehghani committed 5 years ago

Verified 4578df43

Also add link to Jax's dot_general doc.

lucasb-eyer committed 5 years ago

Verified 23b3be96

Add link to DotGeneral in DenseGeneral doc.

lucasb-eyer committed 5 years ago

Verified 78945d4c

Flax README typos cleanup.

dbieber committed 5 years ago

fbdf8513

Merge pull request #301 from ayush-1506:batchnorm_error_msg

a-googler committed 5 years ago

484cc116

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

a-googler committed 5 years ago

639821e1

Concat the distributed batchnorm to make it more efficient.

a-googler committed 5 years ago

a3a6f7ba

Export model_name from the MNIST benchmark.

a-googler committed 5 years ago

59e480f7

Export model_name from the MNIST benchmark

a-googler committed 5 years ago

81752d83

Remove loss scaling from README

mohitreddy1996 committed 5 years ago

Verified 95844ce2

Remove loss_scaling flag from imagenet benchmark

mohitreddy1996 committed 5 years ago

Verified 5a1787b3

Merge pull request #304 from google/test_314238108

levskaya committed 5 years ago

Verified 5493b917

fix create vs fill race condition for multihost wmt training.

levskaya committed 5 years ago

a4b095f3

Add more meaningful error message to batchnorm

ashridh committed 5 years ago

6cbc6331

Merge pull request #285 from kshithijiyer/patch-1

avital committed 5 years ago

Verified 8901bc07

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

avital committed 5 years ago

Verified 967103ee

Merge branch 'master' into patch-1

danielsuo committed 5 years ago

Verified ae45f806

Removing license param from setup() in setup.py

kshithijiyer committed 5 years ago

Verified 057a0492

Add *.rej to .gitignore

danielsuo committed 5 years ago

04bcb23d

Merge pull request #298 from danielsuo/dsuo/update-tensorboard-tests

danielsuo committed 5 years ago

Verified c2a63461

Update tensorboard tests

danielsuo committed 5 years ago

4aec10de

Merge pull request #296 from danielsuo:scheduled-sampling

a-googler committed 5 years ago

37b196d4

Fix indent in checkpointing.diff

danielsuo committed 5 years ago

2bcc42ce

Fix typo in distributed-training.diff

danielsuo committed 5 years ago

b0ca3b54

Merge pull request #295 from danielsuo/dsuo/dummy-checkout

avital committed 5 years ago

Verified 135aac81

Remove dead comment

danielsuo committed 5 years ago

a51bcb93

Clean up check_howto_diffs

danielsuo committed 5 years ago

edf385ee

Older