google/flax

Pull Requests Commits

adds recursive_map

Cristian Garcia committed 134 days ago

f7ba5e02

Merge pull request #4897 from google:pytree-guide

a-googler committed 134 days ago

f4bc868a

Merge pull request #4905 from IvyZX:badge

a-googler committed 135 days ago

bfa5d8e0

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

a-googler committed 135 days ago

5e82ac8d

Update README.md

IvyZX committed 135 days ago

Verified 0c75f1c5

Update README badges and remove invalid ones

IvyZX committed 135 days ago

b3212f06

Merge pull request #4919 from hanrach9:main

a-googler committed 135 days ago

d3c472d2

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

a-googler committed 136 days ago

f40d2bb4

Merge pull request #4904 from ywrt:patch-1

a-googler committed 136 days ago

49cfa78c

Added missing functions to graph.rst

vfdev-5 committed 137 days ago

2e9eaf28

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

vfdev-5 committed 140 days ago

fb648ecb

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

vfdev-5 committed 140 days ago

ab2efb97

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

hanrach9 committed 141 days ago

Verified 42a7443f

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

hanrach9 committed 141 days ago

Verified 8b4a5c19

Update performance.ipynb

hanrach9 committed 141 days ago

Verified 987fdba3

Update performance.md

hanrach9 committed 141 days ago

Verified 21ebcef0

Cristian Garcia committed 141 days ago

f73aea5c

Fix typo in unflatten docs

Cristian Garcia committed 141 days ago

db06488e

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

arnoegw committed 141 days ago

08068bec

update version to 0.11.2

Cristian Garcia committed 141 days ago

405f6566

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

a-googler committed 142 days ago

2bf5748d

merge no copy Variables

cgarciae committed 143 days ago

e8283190

Merge pull request #4909 from google:fix-mypy

a-googler committed 143 days ago

485e101d

Fix bug when raising ScopeParamNotFoundError.

saugenst committed 143 days ago

c24e6556

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

arnoegw committed 143 days ago

c45197e3

cgarciae committed 144 days ago

011b69fd

cgarciae committed 144 days ago

1f10b9da

revert is_leaf logic in _check_carry_same_references

Cristian Garcia committed 144 days ago

7eba0111

Update optimizer.py to support masked variable from optax.

ywrt committed 147 days ago

Verified 2f213b54

Merge pull request #4857 from nenuadrian:patch-1

a-googler committed 148 days ago

a1451a0e

Older