google/flax

Pull Requests Commits

Merge pull request #5045 from samanklesaria:issues/5044

a-googler committed 63 days ago

dc7e7088

docstring update

chapman20j committed 63 days ago

25b1a25b

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

a-googler committed 63 days ago

eb38aad7

Fix shard_map documentation link in compilation.py

vfdev-5 committed 63 days ago

5da4f9c2

Merge pull request #5043 from samanklesaria:issues/4426

a-googler committed 63 days ago

df0cddb3

Use in-place update

samanklesaria committed 63 days ago

2a68940c

samanklesaria committed 63 days ago

da5aca49

Prevent child ctxtag from overwriting parent

samanklesaria committed 63 days ago

fb391c11

filter grad state in nnx.Optimizer

Cristian Garcia committed 64 days ago

241456f6

Merge branch 'google:main' into recursive_map

chapman20j committed 64 days ago

Verified b868f304

Internal change

IvyZX committed 64 days ago

3d1422c9

Internal change

IvyZX committed 64 days ago

60c46d2a

Change introduction to randomness

samanklesaria committed 65 days ago

00601f0d

recursive_map commit

chapman20j committed 66 days ago

0d9cd5aa

chapman20j committed 66 days ago

f0c1a561

Copybara import of the project:

a-googler committed 66 days ago

5a2da6d2

Merge pull request #5037 from google:expose-restore_int_paths

a-googler committed 66 days ago

57c2cc2d

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

a-googler committed 66 days ago

4d1cfbb1

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

vfdev-5 committed 67 days ago

2ea9f3b0

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

a-googler committed 67 days ago

bb3c31ac

Merge pull request #5034 from google:fix-reseed

a-googler committed 70 days ago

6770d71a

fix reseed for abstract values

cgarciae committed 70 days ago

0e490a7c

Support for python 3.14

vfdev-5 committed 70 days ago

6622efcf

Merge pull request #5033 from IvyZX:shard-msg

a-googler committed 71 days ago

c0ea12d3

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

a-googler committed 71 days ago

303f8b29

Add flip doc link to eager sharding error message

IvyZX committed 71 days ago

da8ef361

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

a-googler committed 71 days ago

760be244

Fix tf installation on 3.13 and protobuf version

vfdev-5 committed 71 days ago

ede2bb1b

test against python 3.13

cgarciae committed 71 days ago

44b97a8d

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

a-googler committed 72 days ago

befe5718

Newer Older