google/flax

Pull Requests Commits

Add a customizable conv_transpose.

IvyZX committed 110 days ago

8e05d676

Avoid passing non-boolean mask to `where` argument of `jax.numpy` reductions. Non-boolean mask inputs have been deprecated for several releases, and will result in an error starting in JAX v0.8.0.

Jake VanderPlas committed 112 days ago

4887f7dc

Cristian Garcia committed 112 days ago

2e366136

Merge pull request #4897 from google:pytree-guide

a-googler committed 116 days ago

f4bc868a

Merge pull request #4905 from IvyZX:badge

a-googler committed 118 days ago

bfa5d8e0

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

a-googler committed 118 days ago

5e82ac8d

Update README.md

IvyZX committed 118 days ago

Verified 0c75f1c5

Update README badges and remove invalid ones

IvyZX committed 118 days ago

b3212f06

Merge pull request #4919 from hanrach9:main

a-googler committed 118 days ago

d3c472d2

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

a-googler committed 119 days ago

f40d2bb4

Merge pull request #4904 from ywrt:patch-1

a-googler committed 119 days ago

49cfa78c

Added missing functions to graph.rst

vfdev-5 committed 120 days ago

2e9eaf28

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

vfdev-5 committed 123 days ago

fb648ecb

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

vfdev-5 committed 123 days ago

ab2efb97

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

hanrach9 committed 123 days ago

Verified 42a7443f

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

hanrach9 committed 123 days ago

Verified 8b4a5c19

Update performance.ipynb

hanrach9 committed 123 days ago

Verified 987fdba3

Update performance.md

hanrach9 committed 123 days ago

Verified 21ebcef0

Cristian Garcia committed 123 days ago

f73aea5c

Fix typo in unflatten docs

Cristian Garcia committed 124 days ago

db06488e

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

arnoegw committed 124 days ago

08068bec

update version to 0.11.2

Cristian Garcia committed 124 days ago

405f6566

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

a-googler committed 125 days ago

2bf5748d

merge no copy Variables

cgarciae committed 125 days ago

e8283190

Merge pull request #4909 from google:fix-mypy

a-googler committed 126 days ago

485e101d

Fix bug when raising ScopeParamNotFoundError.

saugenst committed 126 days ago

c24e6556

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

arnoegw committed 126 days ago

c45197e3

cgarciae committed 126 days ago

011b69fd

cgarciae committed 126 days ago

1f10b9da

revert is_leaf logic in _check_carry_same_references

Cristian Garcia committed 126 days ago

7eba0111

Older