google/flax

Pull Requests Commits

Test (please ignore)

marcvanzee committed 5 years ago

Verified 425c834c

cghawthorne committed 5 years ago

5962b295

Merge pull request #863 from SamuelMarks:typo

a-googler committed 5 years ago

e44c89ea

Merge pull request #764 from 8bitmp3/annotated

avital committed 5 years ago

Verified dcbe4721

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

a-googler committed 5 years ago

bb215c88

Merge pull request #947 from levskaya:msgpackfix

a-googler committed 5 years ago

09a7c91c

Add automatic giant array chunking in msgpack checkpoints.

levskaya committed 5 years ago

4ecb68ad

Merge pull request #951 from cleichner:patch-1

a-googler committed 5 years ago

59244847

Fix type in ImageNet/Resnet Example

cleichner committed 5 years ago

Verified b00bc1ef

Fix getattr unbound bug

jheek committed 5 years ago

d9604918

Merge pull request #910 from google:wmt-jit

a-googler committed 5 years ago

f1a1be11

Updates comment on jit(init)

marcvanzee committed 5 years ago

7795d866

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

a-googler committed 5 years ago

ccd2f103

Merge pull request #938 from levskaya:lazy_setup

a-googler committed 5 years ago

836946a6

Call setup lazily.

levskaya committed 5 years ago

46579d13

Ensure random.fold_in receives int32 input.

a-googler committed 5 years ago

58c62fae

allow passing a specific checkpoint path to restore_checkpoint.

cghawthorne committed 5 years ago

2a0a9e2c

allow passing a specific checkpoint path to restore_checkpoint.

cghawthorne committed 5 years ago

1768cb53

Fix WMT example checkpointing frequency and checkpoint from the host.

a-googler committed 5 years ago

6deb7039

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

cghawthorne committed 5 years ago

c289bf11

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

a-googler committed 5 years ago

435a188f

Merge pull request #871 from google:marcvanzee-patch-1

a-googler committed 5 years ago

e533f664

Update comment on sync_batch_stats

marcvanzee committed 5 years ago

f4d234b6

Adds profiler.collect() and makes Profile(logdir) mandatory.

andsteing committed 5 years ago

7cdfcafe

Merge pull request #937 from levskaya:transform_improvements2

a-googler committed 5 years ago

38323b2b

Generalize transform handling for nested attribute submodules.

levskaya committed 5 years ago

cf2a2893

Merge pull request #931 from google:marcvanzee-patch-5

a-googler committed 5 years ago

8f9f7ece

Adds comment to "focus" of OptimizerDef.create()

marcvanzee committed 5 years ago

Verified 5b90b7d1

Merge pull request #925 from avital:just-sphinx-2

a-googler committed 5 years ago

30b21fa5

Use doctest for new HOWTOs, instead of our old diff system

avital committed 5 years ago

265afe1f

Older