google/flax

Pull Requests Commits

WIP adding migrations guides

vfdev-5 committed 166 days ago

6ff94fa4

WIP added migrations guides

vfdev-5 committed 166 days ago

d80b8c9c

Merge pull request #4822 from google:add-train-script-gemma-example

a-googler committed 167 days ago

97f4a499

Merge pull request #4815 from lukeyeh:push-tpqklpwqzxkq

a-googler committed 168 days ago

db0e3020

Added training script for Gemma model

vfdev-5 committed 173 days ago

2eb7baa1

Merge pull request #4820 from google:nnx-fix-rng-dtype-check

a-googler committed 173 days ago

8447c032

Merge pull request #4823 from google:fix-trailing-whitespaces

a-googler committed 173 days ago

4bea978c

Fixed .type deprecated property usage

vfdev-5 committed 173 days ago

539d9f01

Fix trailing whitespace

vfdev-5 committed 173 days ago

f179e8ee

Merge pull request #4800 from google:nnx-remove-variable-state

a-googler committed 174 days ago

2e779f57

Allow using LazyRngs for flax init/apply.

levskaya committed 174 days ago

1ccd1598

[nnx] fix Rngs dtype check

cgarciae committed 174 days ago

164c6be2

Merge pull request #4814 from divyashreepathihalli:use-id

a-googler committed 175 days ago

e147958f

refactor: move usages of `.value` to `[...]` in modules_test.py

lukeyeh committed 175 days ago

9acdcfdc

divyashreepathihalli committed 176 days ago

a587aaca

Merge pull request #4813 from google:to-linen-eval-shape-2

a-googler committed 178 days ago

8c22f295

modify NNX to use id(variable) instead of nnx.Variables as dictionary keys

divyashreepathihalli committed 178 days ago

f411a322

Merge pull request #4776 from google:add-note-gemma-guide-about-model-license

a-googler committed 179 days ago

23d1a1b4

Merge pull request #4790 from google:enable-docs-nnx-remat

a-googler committed 179 days ago

a3f5be21

[nnx] ToLinen add abtract_init flag

cgarciae committed 179 days ago

7268e5f5

Merge pull request #4809 from IvyZX:doc2

a-googler committed 180 days ago

188baabc

Deduplicate contributing and philosophy and move to main site

IvyZX committed 180 days ago

103f65de

Preserve sharding information in axes_scan

oliverdutton-iso committed 181 days ago

4e2becd8

Internal change

kho committed 181 days ago

cbb4f80f

Merge pull request #4808 from google:to-linen-method-calls

a-googler committed 181 days ago

171b28e8

[nnx] allow method calls in ToLinen

cgarciae committed 181 days ago

61d0e9e0

[nnx] remove VariableState

cgarciae committed 184 days ago

e8984b70

Merge pull request #4755 from google:nnx-remove-all

a-googler committed 188 days ago

99f8e598

Merge pull request #4798 from google:ver-0.10.7

a-googler committed 188 days ago

1ecd34f9

update version to 0.10.7

cgarciae committed 188 days ago

d2cf8c2b

Older