google/flax

Pull Requests Commits

Run doctests on GitHub actions

avital committed 4 years ago

6a561c65

Updated Module docstring

joelgarde committed 4 years ago

Verified 1749800d

Adds pydoc for converting collections (& fixes test).

andsteing committed 4 years ago

0d0d8094

Adds missing convert_pre_linen() for batch_stats.

andsteing committed 4 years ago

f4cefee5

Adds test_convert_checkpoint().

andsteing committed 4 years ago

f964aa4f

Linen LM1B training example based on WMT

jaehlee committed 4 years ago

7d12d20f

Adds checkpoints.convert_pre_linen()

andsteing committed 4 years ago

091892a1

No label smoothing for evaluation and fixes a typo in input_pipeline.

jaehlee committed 4 years ago

c86c8b92

Merge pull request #1031 from jheek:no-enforce-frozen

a-googler committed 4 years ago

3e649926

Bumps jaxlib to 0.1.59 in examples/cloud/startup_script.

andsteing committed 4 years ago

8459aef0

Merge pull request #845 from jheek:is-mutable

a-googler committed 4 years ago

2d00aade

Do not enforce collections passed to apply to be frozen

jheek committed 4 years ago

60ad50cb

Merge pull request #1027 from levskaya:8bitmp3-numpy-convention

a-googler committed 4 years ago

aaec8ef6

Merge pull request #913 from yang-song:patch-1

a-googler committed 4 years ago

55877c3b

add an entry in CHANGELOG

yang-song committed 4 years ago

edb30810

Keep the behavior consistent when mutable is an empty dictionary

yang-song committed 4 years ago

e36d4af4

Merge branch 'numpy-convention' of https://github.com/8bitmp3/flax into 8bitmp3-numpy-convention

levskaya committed 4 years ago

3f8cec69

Rename across all of Flax: onp -> np for ordinary NumPy

8bitmp3 committed 4 years ago

634a78ff

Remove NumPy as onp (unused) in flax/optim/dynamic_scale.py

8bitmp3 committed 4 years ago

2f5c00b5

Remove unused ordinary NumPy import in examples/ppo/agent.py (old API)

8bitmp3 committed 4 years ago

e2141424

Remove mentions of unused ordinary NumPy

8bitmp3 committed 4 years ago

2014589e

Merge pull request #1001 from joaogui1:deprecation

a-googler committed 4 years ago

25a3ffb1

Merge pull request #1024 from jheek:vmap-axis_name

a-googler committed 4 years ago

33581564

Merge pull request #1023 from jheek:fix-module-attr-wrapping

a-googler committed 4 years ago

149018f6

Add axis_name arg to lifted vmap

jheek committed 4 years ago

0d2251aa

explain warning line

joaogui1 committed 4 years ago

a07f35a2

joaogui1 committed 4 years ago

d43acdab

Apply suggestions from code review

joaogui1 committed 4 years ago

Verified 4bee3d55

Fix incorrect attribute wrapping

jheek committed 4 years ago

145cb053

Merge pull request #976 from jheek:module-pass-through-attr

a-googler committed 4 years ago

d5a7fe1d

Newer Older