google/flax

Pull Requests Commits

Adds CLU metric logging.

andsteing committed 4 years ago

b11499ae

Makes seq2seq use Optax.

andsteing committed 4 years ago

0d2b116a

Removes deprecated @jax.mask transformation.

andsteing committed 4 years ago

35e44b0c

Merge pull request #1372 from andsteing:imagenet

a-googler committed 4 years ago

b8f1c272

Add flax.png for the sidebar

8bitmp3 committed 4 years ago

Verified cc9636a0

Add html_logo to conf.py for a sidebar logo

8bitmp3 committed 4 years ago

Verified 422e331c

Merge pull request #1379 from jacobhepkema/patch-1

jheek committed 4 years ago

Verified 3abeab0e

Correct typo: n --> np (numpy)

jacobhepkema committed 4 years ago

Verified 8b05bc9c

[flax] Fix a string formatting bug, and some other minor issues.

a-googler committed 4 years ago

074ec83e

[flax] Add a `min_pool` function.

a-googler committed 4 years ago

1470dbdc

Updates imagenet Colab & checkpoints for Optax.

andsteing committed 4 years ago

635c79b7

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

Newer Older