google/flax

Pull Requests Commits

Add MLPerf Transformer to Flax examples

jekbradbury committed 5 years ago

af387cd1

Merge pull request #643 from jheek:add-pytype

a-googler committed 5 years ago

b9f57b4e

Add test case for PixelCNN to run a single training and evaluation step.

Marvin182 committed 5 years ago

d372a896

Merge pull request #671 from andsteing:imagenet_structure

a-googler committed 5 years ago

ce24acc4

Removes --data_dir, renames --model_dir to --workdir.

andsteing committed 5 years ago

90022e2a

Merge pull request #662 from psc-g:patch-1

a-googler committed 5 years ago

e44c5a38

Merge pull request #655 from jheek:frozen-dict-perf

a-googler committed 5 years ago

a2f8a3d6

Renames files according to updated #231.

andsteing committed 5 years ago

a51ba6a2

Adding tests for centered computation

psc-g committed 5 years ago

da79af33

Trains imagenet for 100 epochs by default.

andsteing committed 5 years ago

c3b828a2

Adds imagenet requirements.txt

andsteing committed 5 years ago

f054d232

Adds --data_dir.

andsteing committed 5 years ago

888b84e4

Updates training time in README.

andsteing committed 5 years ago

e601ce0d

jheek committed 5 years ago

dd360c91

Fix cloning non-init fields

jheek committed 5 years ago

b5c70daf

add PyTreeNode docstring

jheek committed 5 years ago

1c9e0f5b

Remove import hack

jheek committed 5 years ago

c6175f1d

Remove comment about generics

jheek committed 5 years ago

373a954a

use type instead of Type

jheek committed 5 years ago

291a5f65

jheek committed 5 years ago

a0b16b05

jheek committed 5 years ago

692b960a

jheek committed 5 years ago

0b1d414a

Merge pull request #664 from 8bitmp3/patch-1

andsteing committed 5 years ago

Verified 0b5785e3

Merge pull request #666 from afqueiruga:master

a-googler committed 5 years ago

4279b82e

Merge pull request #1 from afqueiruga/afqueiruga-patch-1

afqueiruga committed 5 years ago

Verified 30a91470

Fixed typos and missing attribute in module docstrings.

afqueiruga committed 5 years ago

Verified c02ab475

Minor linting - Flax basics

8bitmp3 committed 5 years ago

Verified 47620add

Merge branch 'patch-1' of github.com:psc-g/flax into patch-1

psc-g committed 5 years ago

2969cb5b

Updating RMSProp tests to include param.

psc-g committed 5 years ago

3ebc3e97

Add the centered version of RMSProp

psc-g committed 5 years ago

Verified b209296f

Older