google/flax

Pull Requests Commits

fixed argument 'config' bug

a-googler committed 4 years ago

2b94abcd

Merge pull request #1371 from andsteing/examples

andsteing committed 4 years ago

Verified 87a21113

Adds google/vision_transformer example link.

andsteing committed 4 years ago

ad6a4aed

Merge pull request #1368 from levskaya:unhashable_exception

a-googler committed 4 years ago

e4b29536

Change unhashable module exception type to TypeError.

levskaya committed 4 years ago

80a562c6

Cleaning up formatting of Flax WMT example README.

znado committed 4 years ago

82cc2657

Merge pull request #1365 from levskaya:moduletyping

a-googler committed 4 years ago

31965233

Merge pull request #1358 from andsteing:optax

a-googler committed 4 years ago

3109add8

Removes trailing newline.

andsteing committed 4 years ago

4eb4bb8f

Small change to annotated_mnist.

andsteing committed 4 years ago

b4d8226a

Adds note about computing x-ent.

andsteing committed 4 years ago

70d49ea3

Merge remote-tracking branch 'upstream/master' into HEAD

andsteing committed 4 years ago

8b921617

Update flax/optim/__init__.py

andsteing committed 4 years ago

Verified c874b6f3

Update flax/optim/__init__.py

andsteing committed 4 years ago

Verified 0f4947f2

Update flax/optim/__init__.py

andsteing committed 4 years ago

Verified 7540fafc

Merge pull request #1361 from andsteing:optax2

a-googler committed 4 years ago

70d6840c

Migrate jax.host_id to jax.process_index and jax.host_count to jax.process_count.

tutmann committed 4 years ago

dc62cb8e

Add typing stub for Module __call__

levskaya committed 4 years ago

78d999cd

Merge pull request #1362 from ilopezgp:spelling_mnist

a-googler committed 4 years ago

752a2c07

Fix spelling in MNIST Colab

ilopezgp committed 4 years ago

c0f8023a

Use optax's losses and schedules in mnist and imagenet flax examples.

mtthss committed 4 years ago

44ee6f2f

Makes examples/vae use Optax.

andsteing committed 4 years ago

0886961f

Updates "Annotated MNIST" & links to index.rst

andsteing committed 4 years ago

45f4edaf

Updates docs for Optax.

andsteing committed 4 years ago

661c1663

Merge pull request #1354 from jheek:fix-incorrect-named-tuple-conversion

a-googler committed 4 years ago

88464610

Merge pull request #1351 from lucasb-eyer:patch-7

a-googler committed 4 years ago

4548bbdb

Fix NamedTuples being converted to tuples in Module

jheek committed 4 years ago

6759522a

Merge pull request #1350 from google/andsteing-patch-1

jheek committed 4 years ago

Verified 5ebf14f2

lucasb-eyer committed 4 years ago

Verified 25804a14

Adds reference to matthias-wright/flaxmodels

andsteing committed 4 years ago

Verified 7392f1e4

Older