google/flax

Pull Requests Commits

Fix typo: `impost` to `import`

Vilin97 committed 1 year ago

a792633d

Update graph.rst

8bitmp3 committed 1 year ago

Verified 149144e0

Add flatten to nnx.graph autosummary in graph.rst

8bitmp3 committed 1 year ago

Verified 34b8c6b0

[nnx] add cloudpickle support

cgarciae committed 1 year ago

c55cd448

Merge pull request #4250 from google:enable-notebook-doctest

a-googler committed 1 year ago

5d314528

Update index.rst

ariG23498 committed 1 year ago

Verified fcaa303f

enable doctest on notebooks

cgarciae committed 1 year ago

146d2ff3

Merge pull request #4248 from 8bitmp3:fix-links

a-googler committed 1 year ago

2d64500c

Merge pull request #4244 from IvyZX:nnx-landing

a-googler committed 1 year ago

a8bca5d2

add checkpointing guide

IvyZX committed 1 year ago

742c9262

Nitting and adding links

8bitmp3 committed 1 year ago

ac3fafba

Landing page, glossary and misc

IvyZX committed 1 year ago

774ab4e9

Merge pull request #4173 from 8bitmp3:upgrade-nnx-basics

a-googler committed 1 year ago

6469c5df

Merge pull request #4239 from 8bitmp3:upgrade-nnx-scale-up

a-googler committed 1 year ago

7c5afdbf

Upgrade Flax NNX Basics doc

8bitmp3 committed 1 year ago

fe61e7ed

Update Flax NNX Scale Up SPMD guide

8bitmp3 committed 1 year ago

2bf7f9d1

Copybara import of the project:

Cristian Garcia committed 1 year ago

67e2917d

Internal change

IvyZX committed 1 year ago

0b340685

Merge pull request #4242 from IvyZX:doc-redirects

a-googler committed 1 year ago

9c162ab2

Make dummy redirects for all Linen guide paths in NNX site scope

IvyZX committed 1 year ago

56a26952

Merge pull request #4241 from google:update-banner-2

a-googler committed 1 year ago

e0dafb9f

Merge pull request #4229 from wenscarl:scale_type_fix

a-googler committed 1 year ago

b9bbc986

cgarciae committed 1 year ago

bd6ae7c7

Merge pull request #4238 from google:update-banner

a-googler committed 1 year ago

96b0edc8

fix scale dtype and refactor q_dot_dq

wenscarl committed 1 year ago

5edc65a8

cgarciae committed 1 year ago

f706f275

Merge pull request #4237 from google:change-banner

a-googler committed 1 year ago

078b6dd1

point nnx banner to flax-linen

cgarciae committed 1 year ago

ce20ed52

Merge pull request #4236 from google:nnx-improve-docs-3

a-googler committed 1 year ago

ef5248b4

[nnx] improve docs

cgarciae committed 1 year ago

cac222ae

Newer Older