google/flax

Pull Requests Commits

Add notebooks doctest back and unpin orbax-checkpoint

IvyZX committed 1 year ago

f63bd25a

Fix devcontainer setup

jorisSchaller committed 1 year ago

Verified cc7fe9f2

Update Flax NNX Scale Up guide

8bitmp3 committed 1 year ago

7a53026e

update version to 0.10.0

cgarciae committed 1 year ago

39372b94

[nnx] improve init_cache docs

cgarciae committed 1 year ago

363f3dfa

[nnx] state filters

cgarciae committed 1 year ago

66bbc7e2

Merge branch 'google:main' into rnn_nnx

zinccat committed 1 year ago

Verified 8c86f8ec

zinccat committed 1 year ago

8d8073d9

fix rng position, rng carry stream, scan arg

zinccat committed 1 year ago

b19772eb

Merge pull request #4281 from google:improve-randomness-guide

a-googler committed 1 year ago

0e5a0b4d

[nnx] improve randomness guide

cgarciae committed 1 year ago

dc7ef1f5

Fixed nnx examples bad links in the README.md

vfdev-5 committed 1 year ago

Verified 5dc792f3

[nnx] add PrefixMapping

cgarciae committed 1 year ago

39565d35

Merge pull request #4277 from 8bitmp3:update-mnist

a-googler committed 1 year ago

f425c2fa

Merge pull request #4262 from 8bitmp3:fixes-nnx-1

a-googler committed 1 year ago

53ba626c

Update Flax NNX MNIST tutorial

8bitmp3 committed 1 year ago

62b3bbcc

Update Why Flax NNX doc

8bitmp3 committed 1 year ago

50df8097

Merge pull request #4276 from 8bitmp3:update-model-surgery

a-googler committed 1 year ago

8a4d6d6c

Update Flax NNX Model Surgery

8bitmp3 committed 1 year ago

f7785f94

Merge pull request #4274 from 8bitmp3:update-landing

a-googler committed 1 year ago

bde5150f

Update Why Flax NNX doc

8bitmp3 committed 1 year ago

28a423ca

Update Flax NNX Landing Page

8bitmp3 committed 1 year ago

451256c1

Update Flax NNX Model Surgery

8bitmp3 committed 1 year ago

8ca8131a

Update PyO3 to 0.21.2

a-googler committed 1 year ago

fec32840

Merge pull request #4183 from google:nnx-fsdp-toy-example

a-googler committed 1 year ago

2ad97319

[nnx] add nnx.variables + FSDP toy example

cgarciae committed 1 year ago

f7d3873b

fix seq_length in rnn, added test for rnn

zinccat committed 1 year ago

d94b5a37

added recurrent and test for lstmcell

zinccat committed 1 year ago

8cd87ee1

Merge pull request #4255 from 8bitmp3:add-graph-flatten-autosummary

a-googler committed 1 year ago

9f52f662

Merge pull request #4267 from google:remove-flax-nnx-urls

a-googler committed 1 year ago

2a2187bd

Newer Older