google/flax

Pull Requests Commits

Improve error message with class name.

a-googler committed 113 days ago

712eb7ec

Merge pull request #4897 from google:pytree-guide

a-googler committed 118 days ago

f4bc868a

Merge pull request #4905 from IvyZX:badge

a-googler committed 119 days ago

bfa5d8e0

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

a-googler committed 119 days ago

5e82ac8d

Update README.md

IvyZX committed 119 days ago

Verified 0c75f1c5

Update README badges and remove invalid ones

IvyZX committed 119 days ago

b3212f06

Merge pull request #4919 from hanrach9:main

a-googler committed 119 days ago

d3c472d2

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

a-googler committed 120 days ago

f40d2bb4

Merge pull request #4904 from ywrt:patch-1

a-googler committed 120 days ago

49cfa78c

Added missing functions to graph.rst

vfdev-5 committed 121 days ago

2e9eaf28

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

vfdev-5 committed 124 days ago

fb648ecb

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

vfdev-5 committed 124 days ago

ab2efb97

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

hanrach9 committed 125 days ago

Verified 42a7443f

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

hanrach9 committed 125 days ago

Verified 8b4a5c19

Update performance.ipynb

hanrach9 committed 125 days ago

Verified 987fdba3

Update performance.md

hanrach9 committed 125 days ago

Verified 21ebcef0

Cristian Garcia committed 125 days ago

f73aea5c

Fix typo in unflatten docs

Cristian Garcia committed 125 days ago

db06488e

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

arnoegw committed 125 days ago

08068bec

update version to 0.11.2

Cristian Garcia committed 125 days ago

405f6566

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

a-googler committed 126 days ago

2bf5748d

merge no copy Variables

cgarciae committed 127 days ago

e8283190

Merge pull request #4909 from google:fix-mypy

a-googler committed 127 days ago

485e101d

Fix bug when raising ScopeParamNotFoundError.

saugenst committed 127 days ago

c24e6556

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

arnoegw committed 127 days ago

c45197e3

cgarciae committed 128 days ago

011b69fd

cgarciae committed 128 days ago

1f10b9da

revert is_leaf logic in _check_carry_same_references

Cristian Garcia committed 128 days ago

7eba0111

Update optimizer.py to support masked variable from optax.

ywrt committed 131 days ago

Verified 2f213b54

Merge pull request #4857 from nenuadrian:patch-1

a-googler committed 132 days ago

a1451a0e

Older