google/flax

Pull Requests Commits

Test (please ignore)

marcvanzee committed 4 years ago

Verified d8ca188b

Merge pull request #954 from avital:minor-doc-improvement

a-googler committed 4 years ago

4f846809

Merge pull request #949 from google:improve-error

a-googler committed 4 years ago

2e2d6b02

Merge branches 'master' and 'improve-error' of https://github.com/google/flax into improve-error

marcvanzee committed 4 years ago

82b3e5b8

Merge pull request #956 from google/test_354507006

marcvanzee committed 4 years ago

Verified ebc7b30e

Remove image from Imagenet Colab temporarily since it gave an internal problem.

marcvanzee committed 4 years ago

293b1f62

cghawthorne committed 4 years ago

5962b295

Merge pull request #863 from SamuelMarks:typo

a-googler committed 4 years ago

e44c89ea

Merge pull request #764 from 8bitmp3/annotated

avital committed 4 years ago

Verified dcbe4721

Minor README updates

avital committed 4 years ago

803a6733

Merge pull request #950 from jheek:fix-getattr-unbound-bug

a-googler committed 4 years ago

bb215c88

Updates after comments PR

marcvanzee committed 4 years ago

97885b72

Merge branches 'master' and 'improve-error' of https://github.com/google/flax into improve-error

marcvanzee committed 4 years ago

d99313f4

Merge pull request #947 from levskaya:msgpackfix

a-googler committed 4 years ago

09a7c91c

Add automatic giant array chunking in msgpack checkpoints.

levskaya committed 4 years ago

4ecb68ad

Merge pull request #951 from cleichner:patch-1

a-googler committed 4 years ago

59244847

Fix type in ImageNet/Resnet Example

cleichner committed 4 years ago

Verified b00bc1ef

Fix getattr unbound bug

jheek committed 4 years ago

d9604918

Merge pull request #910 from google:wmt-jit

a-googler committed 4 years ago

f1a1be11

Updates comment on jit(init)

marcvanzee committed 4 years ago

7795d866

Merge pull request #908 from google:marcvanzee-patch-4

a-googler committed 4 years ago

ccd2f103

First try-out for better error messages

marcvanzee committed 4 years ago

8910cbd1

Merge pull request #938 from levskaya:lazy_setup

a-googler committed 4 years ago

836946a6

Call setup lazily.

levskaya committed 4 years ago

46579d13

Ensure random.fold_in receives int32 input.

a-googler committed 4 years ago

58c62fae

allow passing a specific checkpoint path to restore_checkpoint.

cghawthorne committed 4 years ago

2a0a9e2c

allow passing a specific checkpoint path to restore_checkpoint.

cghawthorne committed 4 years ago

1768cb53

Fix WMT example checkpointing frequency and checkpoint from the host.

a-googler committed 4 years ago

6deb7039

Move latest_checkpoint functionality to its own method, similar to tf.train.latest_checkpoint.

cghawthorne committed 4 years ago

c289bf11

Merge pull request #872 from google:marcvanzee-patch-2

a-googler committed 4 years ago

435a188f

Older