google/flax

Pull Requests Commits

not actually a public commit, just has a diffbase :/

a-googler committed 4 years ago

a081ef03

Merge pull request #1442 from google/BertrandRdp-tutorial-patch

BertrandRdp committed 4 years ago

Verified 7db4ad2f

Fixing type in JAX tutorial

BertrandRdp committed 4 years ago

Verified 5c01d64e

Merge pull request #1432 from jheek:fix-namedtuple-restore

a-googler committed 4 years ago

0c9208f1

Add changelog entry

jheek committed 4 years ago

1503d918

Add backward compatibility

jheek committed 4 years ago

9e691ae6

Rename more master to main in code.

andsteing committed 4 years ago

b1ebdc87

Fix serialization restore.

jheek committed 4 years ago

bfca2cc2

Merge pull request #1428 from lc0:gde-tests

a-googler committed 4 years ago

f8dc89a6

Fix the default branch

lc0 committed 4 years ago

714569d0

lc0 committed 4 years ago

3e889d93

Remove space at end of line.

gnecula committed 4 years ago

ff7976d8

Merge pull request #1426 from gnecula:rename_test

a-googler committed 4 years ago

f96d1ea8

Rename master to main.

gnecula committed 4 years ago

d909af5f

Merge pull request #1425 from gnecula/rename_master

gnecula committed 4 years ago

Verified 1844de4d

Renaming master to main. Step 1

gnecula committed 4 years ago

b764497e

Merge pull request #1423 from google:marcvanzee-patch-10

a-googler committed 4 years ago

e08cd70e

Add test for frozendict copy

marcvanzee committed 4 years ago

6feac8e5

Add test for frozendict copy

marcvanzee committed 4 years ago

9f31066e

Add test for frozendict copy

marcvanzee committed 4 years ago

e54a9b6d

Update frozen_dict.py

marcvanzee committed 4 years ago

Verified 0a7ca34c

Merge pull request #1415 from vasudevgupta7/bigbird-tutorial

marcvanzee committed 4 years ago

Verified a1a73eb9

Merge pull request #1404 from jheek:ppo-optax

a-googler committed 4 years ago

25c1fdf2

add bigbird example

thevasudevgupta committed 4 years ago

78ffcd44

Refactor based on review

jheek committed 4 years ago

934a8183

Merge pull request #1412 from navjotts/annonated-mnist-correction

andsteing committed 4 years ago

Verified 5c677ea4

use the existing train_ds_size variable throughout

navjotts committed 4 years ago

1e51ad8c

Add data_transform argument to nn.scan and preserve PartitionSpec attributes.

levskaya committed 4 years ago

5ffd0006

Merge pull request #1392 from 8bitmp3/logo-ish

jheek committed 4 years ago

Verified 03f8e7b8

Use Optax optimizer for ppo example

jheek committed 4 years ago

6bab1bf9

Older