google/flax

Pull Requests Commits

cgarciae committed 1 year ago

dd662191

adds recursive_map

Cristian Garcia committed 1 year ago

54a8103f

Merge pull request #4897 from google:pytree-guide

a-googler committed 1 year ago

f4bc868a

Merge pull request #4905 from IvyZX:badge

a-googler committed 1 year ago

bfa5d8e0

Merge pull request #4920 from google:add-preferred_element_type-arg

a-googler committed 1 year ago

5e82ac8d

Update README.md

IvyZX committed 1 year ago

Verified 0c75f1c5

Update README badges and remove invalid ones

IvyZX committed 1 year ago

b3212f06

Merge pull request #4919 from hanrach9:main

a-googler committed 1 year ago

d3c472d2

Merge pull request #4922 from google:update-nnx-graph-docs

a-googler committed 1 year ago

f40d2bb4

Merge pull request #4904 from ywrt:patch-1

a-googler committed 1 year ago

49cfa78c

Added missing functions to graph.rst

vfdev-5 committed 1 year ago

2e9eaf28

Merge branch 'main' of github.com:google/flax into patch-1

vfdev-5 committed 1 year ago

fb648ecb

Added preferred_element_type arg to nnx.Linear*, nnx.Conv*, nnx.Einsum

vfdev-5 committed 1 year ago

ab2efb97

Merge pull request #2 from hanrach9/hanrach9-patch-1-1

hanrach9 committed 1 year ago

Verified 42a7443f

Merge pull request #1 from hanrach9/hanrach9-patch-1

hanrach9 committed 1 year ago

Verified 8b4a5c19

Update performance.ipynb

hanrach9 committed 1 year ago

Verified 987fdba3

Update performance.md

hanrach9 committed 1 year ago

Verified 21ebcef0

Cristian Garcia committed 1 year ago

f73aea5c

Fix typo in unflatten docs

Cristian Garcia committed 1 year ago

db06488e

Doc fix: remove dead link to pre-Orbax checkpointing.

arnoegw committed 1 year ago

08068bec

update version to 0.11.2

Cristian Garcia committed 1 year ago

405f6566

Merge pull request #4912 from google:no-copy-merge

a-googler committed 1 year ago

2bf5748d

merge no copy Variables

cgarciae committed 1 year ago

e8283190

Merge pull request #4909 from google:fix-mypy

a-googler committed 1 year ago

485e101d

Fix bug when raising ScopeParamNotFoundError.

saugenst committed 1 year ago

c24e6556

Doc fix: remove outdated advice on flax v0.6.10; it was released two years ago.

arnoegw committed 1 year ago

c45197e3

cgarciae committed 1 year ago

011b69fd

cgarciae committed 1 year ago

1f10b9da

revert is_leaf logic in _check_carry_same_references

Cristian Garcia committed 1 year ago

7eba0111

Update optimizer.py to support masked variable from optax.

ywrt committed 1 year ago

Verified 2f213b54

Older