google/flax

Pull Requests Commits

Make save_checkpoint more robust to non normal paths.

a-googler committed 4 years ago

34cc1e92

Rename more master to main in code.

andsteing committed 4 years ago

b1ebdc87

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

Merge pull request #1403 from levskaya:generic_fix

a-googler committed 4 years ago

97cd5bc8

Make nn.Module compatible with multiple inheritance with Generic[T].

levskaya committed 4 years ago

2d13147c

Merge pull request #1396 from andsteing:examples

a-googler committed 4 years ago

34eab18d

Adds requirements.

andsteing committed 4 years ago

2fa188a6

andsteing committed 4 years ago

2cbebb76

Adds minimal Colab & updates README.

andsteing committed 4 years ago

a5b2bc9d

Older