google/flax

Pull Requests Commits

Stackless yashful

dougalm committed 1 year ago

7b4a7f54

Merge pull request #4317 from google:improve-state-loading

a-googler committed 1 year ago

8360b7c2

Merge pull request #4289 from 8bitmp3:update-evolution-linen-to-nn

a-googler committed 1 year ago

092ba1d2

Merge pull request #4316 from google:improve-mnist-tutorial-2

a-googler committed 1 year ago

685b255f

[nnx] improve mnist tutorial

cgarciae committed 1 year ago

d2606ce9

[nnx] try casting integers keys in State.replace_by_pure_dict

cgarciae committed 1 year ago

059743c6

Merge pull request #4305 from IvyZX:haiku-nnx

a-googler committed 1 year ago

c1046ad0

Avoid float32 promotion in BatchNorm when using_running_average=True and force_float32_reductions=False

eunjaekim-0 committed 1 year ago

a7f32a42

Update Flax Evolution from Linen to NNX guide

8bitmp3 committed 1 year ago

f00917b9

Merge pull request #4302 from 8bitmp3:add-graphdef-docstring

a-googler committed 1 year ago

702afbeb

Flesh out the Haiku/Flax guide

IvyZX committed 1 year ago

ba63f4cd

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

Add Flax NNX GraphDef docstring

8bitmp3 committed 1 year ago

74423bc1

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

Older