google/flax

Pull Requests Commits

Implements Adabelief - an adaptive method that modifies the stpesize according

a-googler committed 4 years ago

0d985667

Linen LM1B training example based on WMT

jaehlee committed 4 years ago

7d12d20f

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

Merge pull request #1021 from google:avital-patch-5

a-googler committed 4 years ago

9e098fdd

Link from Linen README to the reference docs

avital committed 4 years ago

Verified edcf9401

Merge pull request #1020 from jheek:fix-init-type-annotation

a-googler committed 4 years ago

1fdf7c58

Fix init/apply return annotation

jheek committed 4 years ago

e36b3f18

Generalize attribute assignment to support passing module instances

jheek committed 4 years ago

9bf71619

Older