google/flax

Pull Requests Commits

docstring update

chapman20j committed 220 days ago

25b1a25b

Merge pull request #5038 from google:vfdev-5-patch-1

a-googler committed 220 days ago

eb38aad7

Fix shard_map documentation link in compilation.py

vfdev-5 committed 220 days ago

5da4f9c2

Merge pull request #5043 from samanklesaria:issues/4426

a-googler committed 220 days ago

df0cddb3

Use in-place update

samanklesaria committed 220 days ago

2a68940c

samanklesaria committed 220 days ago

da5aca49

Prevent child ctxtag from overwriting parent

samanklesaria committed 220 days ago

fb391c11

filter grad state in nnx.Optimizer

Cristian Garcia committed 221 days ago

241456f6

Merge branch 'google:main' into recursive_map

chapman20j committed 221 days ago

Verified b868f304

Internal change

IvyZX committed 221 days ago

3d1422c9

Internal change

IvyZX committed 221 days ago

60c46d2a

Change introduction to randomness

samanklesaria committed 222 days ago

00601f0d

recursive_map commit

chapman20j committed 223 days ago

0d9cd5aa

chapman20j committed 223 days ago

f0c1a561

Copybara import of the project:

a-googler committed 223 days ago

5a2da6d2

Merge pull request #5037 from google:expose-restore_int_paths

a-googler committed 223 days ago

57c2cc2d

Merge pull request #5032 from google:fix-py314-kw-only-dataclasses

a-googler committed 223 days ago

4d1cfbb1

[docs] Exposed more functions/classes in state.rst

vfdev-5 committed 224 days ago

2ea9f3b0

Deduplicate `Variable` nodes in `iter_graph` and eliminate recursion.

a-googler committed 224 days ago

bb3c31ac

Merge pull request #5034 from google:fix-reseed

a-googler committed 227 days ago

6770d71a

fix reseed for abstract values

cgarciae committed 227 days ago

0e490a7c

Support for python 3.14

vfdev-5 committed 227 days ago

6622efcf

Merge pull request #5033 from IvyZX:shard-msg

a-googler committed 228 days ago

c0ea12d3

Merge pull request #5031 from google:docs-nnx-jit-arg-donation

a-googler committed 228 days ago

303f8b29

Add flip doc link to eager sharding error message

IvyZX committed 228 days ago

da8ef361

Merge pull request #4987 from google:python-3.13-support

a-googler committed 228 days ago

760be244

Fix tf installation on 3.13 and protobuf version

vfdev-5 committed 228 days ago

ede2bb1b

test against python 3.13

cgarciae committed 228 days ago

44b97a8d

Merge pull request #5008 from mohsinm-dev:fix-nnx-tabulate-variable-hooks

a-googler committed 229 days ago

befe5718

Merge pull request #5023 from rapsealk:docs/readme-nnx-linear

a-googler committed 229 days ago

c4f2c5c5

Newer Older