google/flax

Pull Requests Commits

Replace flax named_call with upstream version jax.named_call since https://github.com/google/jax/pull/4796 has merged.

a-googler committed 5 years ago

b2eb4fb5

Merge pull request #588 from marcvanzee:documentation

a-googler committed 5 years ago

55d169f3

Update scope comment

marcvanzee committed 5 years ago

660a4901

Merge branch 'master' of https://github.com/google/flax into documentation

marcvanzee committed 5 years ago

0b1916bb

Merge pull request #640 from google:module-docstring

a-googler committed 5 years ago

d6e3d203

moved nlp_seq to linen_examples

bohnetbd committed 5 years ago

1c7f06bb

marcvanzee committed 5 years ago

57b7301d

Update sphinx version so types are shown correctly.

marcvanzee committed 5 years ago

fbaef438

Shorten code for variable initialization.

Marvin182 committed 5 years ago

a5c971b2

Update structure of MNIST example.

Marvin182 committed 5 years ago

0c6a8baf

Internal change.

Marvin182 committed 5 years ago

80422e5f

Merge pull request #625 from google:module-docstring

a-googler committed 5 years ago

0c1332f5

marcvanzee committed 5 years ago

89331313

Minor fixes for readthedocs

marcvanzee committed 5 years ago

bc95a9f2

Merge pull request #626 from google:nlp_seq_linen

a-googler committed 5 years ago

a47c375d

marcvanzee committed 5 years ago

2872ad08

Updates docstring in Module.setup

marcvanzee committed 5 years ago

b9a8da81

Merge branch 'master' into module-docstring

marcvanzee committed 5 years ago

Verified a7d54fac

Incorporated comments Anselm

marcvanzee committed 5 years ago

c8aa65c9

Merge pull request #626 from google/nlp_seq_linen

bohnetbd committed 5 years ago

Verified d9604c1c

Changes due to review.

bohnetbd committed 5 years ago

50ade957

Add missing dependency to WMT example.

Marvin182 committed 5 years ago

60ebf7e2

Merge pull request #603 from google:docs_revamp

a-googler committed 5 years ago

658022d5

Merge pull request #628 from jheek/raise-minimal-version

avital committed 5 years ago

Verified c2afe65f

Raise minimal version to 0.1.77

jheek committed 5 years ago

9468f151

Add separate main.py, switch to ConfigDict and add basic test case.

Marvin182 committed 5 years ago

8ce8e5cd

Merge pull request #602 from marcvanzee:core

a-googler committed 5 years ago

475a2b63

Changes due to review of andsteing no resolved PR #609

bohnetbd committed 5 years ago

6b6d9e20

Merge branch 'master' of https://github.com/google/flax into module-docstring

marcvanzee committed 5 years ago

4ca8141f

Updates comments

marcvanzee committed 5 years ago

bad5ca7a

Older