google/flax

Pull Requests Commits

Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.

cpgaffney1 committed 2 years ago

90373640

Merge pull request #4067 from IvyZX:misc

a-googler committed 2 years ago

0c3c74c8

uncomment some tests after 3.9 dropped

IvyZX committed 2 years ago

e800d5b0

Merge pull request #4036 from kaushaladiti-2802:patch-1

a-googler committed 2 years ago

36612a38

Update orbax_upgrade_guide.rst

kaushaladiti-2802 committed 2 years ago

Verified 49100052

Merge pull request #4065 from google:fix-accuracy

a-googler committed 2 years ago

0da4f31d

Update orbax_upgrade_guide.rst

kaushaladiti-2802 committed 2 years ago

Verified 7298cc5c

[nnx] fix Accuracy on eager mode

cgarciae committed 2 years ago

26d5e68d

Update orbax_upgrade_guide.rst

kaushaladiti-2802 committed 2 years ago

Verified 5406692c

Update orbax_upgrade_guide.rst

kaushaladiti-2802 committed 2 years ago

Verified 8e307f10

Merge pull request #4063 from google:fix-notebook-links

a-googler committed 2 years ago

4e911139

Merge pull request #3984 from kaixih:rename_fp8_custom_dtype

a-googler committed 2 years ago

b3785dc7

[nnx] fix mnist_tutorial colab link

cgarciae committed 2 years ago

0be01f9e

Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.

dubey committed 2 years ago

96cf6490

Merge pull request #4030 from google:nnx-pytree

a-googler committed 2 years ago

96bb44ca

[nnx] stabilize unsafe_pytree

cgarciae committed 2 years ago

4ed137c5

Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.

cpgaffney1 committed 2 years ago

0fb17771

Update orbax_upgrade_guide.rst for async checkpointing usage examples

kaushaladiti-2802 committed 2 years ago

4b4d2fee

Merge pull request #4049 from google:nnx-fix-variable-inplace-operators

a-googler committed 2 years ago

1367acb1

kaixih committed 2 years ago

798cfe7b

Rename the new dtype

kaixih committed 2 years ago

16f28666

Merge pull request #4047 from rajasekharporeddy:main

a-googler committed 2 years ago

3112da6f

Add functool.wraps() annotation to flax.nn.jit.

hawkinsp committed 2 years ago

37123d52

[nnx] fix Variable overloads and add shape/dtype properties

cgarciae committed 2 years ago

385ca738

rajasekharporeddy committed 2 years ago

Verified 956ad6b3

rajasekharporeddy committed 2 years ago

Verified 89fd9f4d

Merge pull request #3993 from wenchenvincent:ocp_nanoo_fp8

a-googler committed 2 years ago

5b8265c6

Merge pull request #3808 from google:linen-allow-multi-compact

a-googler committed 2 years ago

646a6463

Merge pull request #3988 from google:nnx-lora-param

a-googler committed 2 years ago

b5503da5

[linen] allows multiple compact methods

cgarciae committed 2 years ago

1f291978

Older