google/flax

Pull Requests Commits

[nnx] flaxlib flatten

cgarciae committed 1 year ago

239f8f6b

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

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

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

remove flax-nnx urls

cgarciae committed 1 year ago

b12c4b41

Merge pull request #4266 from google:custom-vjp-docs

a-googler committed 1 year ago

601a9150

Merge pull request #4226 from google:nnx-train-state-toy-example

a-googler committed 1 year ago

62a3414f

[nnx] add custom_vjp to docs

cgarciae committed 1 year ago

312368e6

Merge pull request #4256 from Vilin97:main

a-googler committed 1 year ago

40b9b3a8

Merge pull request #4253 from google:cloud-pickle-support

a-googler committed 1 year ago

b487b52e

Merge pull request #4240 from google:why-nnx

a-googler committed 1 year ago

94f6440c

add Why NNX doc

cgarciae committed 1 year ago

caa061d0

Merge pull request #4264 from 8bitmp3:fixes-nnx-2

a-googler committed 1 year ago

11f36163

Add checkpointing guide to website index.

IvyZX committed 1 year ago

bb4d44af

Light update to Flax NNX Transforms doc

8bitmp3 committed 1 year ago

a1543a81

Merge pull request #4249 from IvyZX:ckpt-guide

a-googler committed 1 year ago

9a3a1fc1

Older