google/flax

Pull Requests Commits

[nnx] fix custom_vjp

cgarciae committed 1 year ago

1ef845c4

Merge pull request #4292 from google:v0.10.0

a-googler committed 1 year ago

c6921142

Merge pull request #4284 from 8bitmp3:lint-glossary

a-googler committed 1 year ago

168aa3b1

Merge pull request #4272 from zinccat:rnn_nnx

a-googler committed 1 year ago

3bf732c8

Merge pull request #4296 from 8bitmp3:nit-gspmd

a-googler committed 1 year ago

ae5c6626

Merge pull request #4294 from 8bitmp3:update-nnx-checkpointing

a-googler committed 1 year ago

0bf1f92f

Update Flax NNX Glossary

8bitmp3 committed 1 year ago

46ac8625

Update Flax NNX Glossary

8bitmp3 committed 1 year ago

0978a45d

Ugrade Flax NNX Checkpointing guide

8bitmp3 committed 1 year ago

36a4a70a

Merge pull request #4299 from jorisSchaller:fix_docker

a-googler committed 1 year ago

764e1732

Merge pull request #4288 from google:state-filters

a-googler committed 1 year ago

5322ea54

Merge pull request #4285 from IvyZX:nbunbreak

a-googler committed 1 year ago

feb399db

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] 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

[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

Older