google/flax

Pull Requests Commits

array refs refactor

cgarciae committed 163 days ago

af593059

Merge pull request #4795 from IvyZX:doc

a-googler committed 165 days ago

210e4d6d

Merge pull request #4796 from SobhanMP:identity-sequantial

a-googler committed 165 days ago

ac44f694

Add a sharp bits doc

IvyZX committed 165 days ago

4202efac

Merge pull request #4854 from google:mutable-array-guide

a-googler committed 165 days ago

c9fc00b4

add 0.11.0 migration guide

cgarciae committed 165 days ago

2de53d54

Merge pull request #4841 from lukeyeh:push-kqruztmzotnw

a-googler committed 166 days ago

c429072b

Make `Sequential()` behave like the identity function

SobhanMP committed 169 days ago

10b95ba5

Merge pull request #4851 from google:update-support-mutable-array

a-googler committed 169 days ago

065d6e6b

[nnx] add update support for mutable array

cgarciae committed 169 days ago

d1c1043f

Merge pull request #4842 from google:optimizer-keep-model

a-googler committed 169 days ago

aeed717a

Merge pull request #4842 from google:optimizer-keep-model

a-googler committed 169 days ago

94c94802

create ModelAndOptimizer symbol

Cristian Garcia committed 169 days ago

77196487

[nnx] add Optimizer keep_model param

cgarciae committed 170 days ago

c6498d57

Merge pull request #4811 from google:nnx-pytree-modules-2

a-googler committed 170 days ago

ae208ee7

Migrate `transforms_test.py` from `.value` to `[...]`

lukeyeh committed 172 days ago

726a01dd

[nnx] add flax_pytree_module flag

cgarciae committed 172 days ago

50e043c2

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

a-googler committed 172 days ago

97f4a499

Merge pull request #4815 from lukeyeh:push-tpqklpwqzxkq

a-googler committed 172 days ago

db0e3020

Added training script for Gemma model

vfdev-5 committed 178 days ago

2eb7baa1

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

a-googler committed 178 days ago

8447c032

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

a-googler committed 178 days ago

4bea978c

Fixed .type deprecated property usage

vfdev-5 committed 178 days ago

539d9f01

Fix trailing whitespace

vfdev-5 committed 178 days ago

f179e8ee

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

a-googler committed 179 days ago

2e779f57

Allow using LazyRngs for flax init/apply.

levskaya committed 179 days ago

1ccd1598

[nnx] fix Rngs dtype check

cgarciae committed 179 days ago

164c6be2

Merge pull request #4814 from divyashreepathihalli:use-id

a-googler committed 180 days ago

e147958f

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

lukeyeh committed 180 days ago

9acdcfdc

divyashreepathihalli committed 180 days ago

a587aaca

Older