google/flax

Pull Requests Commits

Merge pull request #1723 from jheek:lazy-rng

a-googler committed 3 years ago

b60f7f45

Merge pull request #1774 from andsteing/optax

andsteing committed 3 years ago

Verified 42c9b604

Applies Marc's remaining comments.

andsteing committed 3 years ago

8d4cf049

Merge pull request #1621 from andsteing/doc

andsteing committed 3 years ago

Verified 7c98266b

Removed extra "test:".

andsteing committed 3 years ago

48a6f201

Applied Marc's suggestions.

andsteing committed 3 years ago

a0369aec

Merge pull request #1793 from melissatan/patch-4

marcvanzee committed 3 years ago

Verified 9bb95f29

melissatan committed 3 years ago

Verified c9073461

Merge pull request #1784 from saeta:fix-license

a-googler committed 3 years ago

7d6844d3

Fix abstract PRNG shape when creating params.

jpuigcerver committed 3 years ago

03f685a9

Fix duplicate licenses in other files too.

saeta committed 3 years ago

f835a55b

Remove duplicate license header.

saeta committed 3 years ago

4d83151c

Merge pull request #1699 from NeilGirdhar/patch-1

levskaya committed 3 years ago

Verified 89e1126c

Two updates suggested by Avital.

andsteing committed 3 years ago

d17c0ae8

Update docs/howtos/optax_update_guide.rst

andsteing committed 3 years ago

Verified a3cbbbcc

Update docs/howtos/optax_update_guide.rst

andsteing committed 3 years ago

Verified aee1dba5

Copybara import of the project:

dependabot[bot] committed 3 years ago

c1abf11b

Merge pull request #1781 from google:dependabot/pip/examples/vae/pillow-9.0.0

a-googler committed 3 years ago

1500af62

validate RNG key shape against jax's default

froystig committed 3 years ago

c82cc742

Bump pillow from 8.3.2 to 9.0.0 in /examples/vae

dependabot[bot] committed 3 years ago

Verified 2771a83e

Adressing @rosshemsley's comments.

andsteing committed 3 years ago

057cdcdb

Adds setup code and cross references LR schedules.

andsteing committed 3 years ago

e25260ff

Adds optax update guide.

andsteing committed 3 years ago

1451cb8b

Handle setup with transformed methods taking submodules of self.

levskaya committed 4 years ago

fb4f2a45

Dmitry Tsarkov committed 4 years ago

b36e873a

Merge pull request #1753 from andsteing:doc2

a-googler committed 4 years ago

63c5d5cc

revert setup transform changes out of caution, may not be generally safe.

levskaya committed 4 years ago

6a2e1b3f

Merge pull request #1745 from levskaya:setup_idempotency

a-googler committed 4 years ago

712e780c

Makes `FLAX_PROFILE=true` gotcha glaringly obvious.

andsteing committed 4 years ago

24ce9616

Copybara import of the project:

dependabot[bot] committed 4 years ago

151dbc1d

Newer Older