google/flax

Pull Requests Commits

Add bridge guide

IvyZX committed 1 year ago

7be92ac3

Merge pull request #4158 from IvyZX:bdg-doc

a-googler committed 2 years ago

559ba024

Add nnx bridge API reference to site

IvyZX committed 2 years ago

93821f7d

Merge pull request #4145 from IvyZX:bdg

a-googler committed 2 years ago

839db8c1

Support linen <-> nnx metadata box converging in bridge

IvyZX committed 2 years ago

4bb2152d

Merge pull request #4149 from google:nnx-fix-nnx-basics-prints

a-googler committed 2 years ago

afbc5021

[nnx] clear nnx basics pip logs

cgarciae committed 2 years ago

366aeff3

update Flax to version 0.9.0

Cristian Garcia committed 2 years ago

fc2e5993

Merge pull request #4148 from google:fix-headers

a-googler committed 2 years ago

1b08425c

remove repeated license headers

cgarciae committed 2 years ago

4be609d7

Merge pull request #4141 from google:nnx-landing-page

a-googler committed 2 years ago

a0622b07

[nnx] improve landing page

cgarciae committed 2 years ago

4e17aa10

Merge pull request #3963 from google:nnx-experimental-transforms

a-googler committed 2 years ago

3a9d8339

Merge pull request #4123 from cool-RR:patch-4

a-googler committed 2 years ago

d0942f9b

Merge pull request #4137 from IvyZX:wrapper-fix

a-googler committed 2 years ago

1b2cca9f

Merge pull request #4136 from google:use-uv

a-googler committed 2 years ago

2ebd3984

use uv package manager

cgarciae committed 2 years ago

6197477e

add uv.lock file

Cristian Garcia committed 2 years ago

cfc4989b

More testing and misc fixes on wrappers

IvyZX committed 2 years ago

a4aa7eb0

Merge pull request #4120 from dcharatan:parent_type_fix

a-googler committed 2 years ago

54634f7e

Merge pull request #4108 from google:nnx-transforms-flip

a-googler committed 2 years ago

9dba6c3f

Merge pull request #4134 from google:fix-object-state

a-googler committed 2 years ago

a6edb8ce

[nnx] fix initializing propagation

cgarciae committed 2 years ago

f6e7962d

Merge pull request #4095 from treigerm:nnx-add-groupnorm

a-googler committed 2 years ago

a3ddb0fa

Switch NNX to use Treescope instead of Penzai.

danieldjohnson committed 2 years ago

7c96b0e1

Fix link to orbax documentation

cool-RR committed 2 years ago

885d8bcf

[nnx] experimental transforms

cgarciae committed 2 years ago

faf9f67f

Merge pull request #4126 from IvyZX:bridge

a-googler committed 2 years ago

71b5a46b

IvyZX committed 2 years ago

fe1422e9

[pmap no rank reduce cleanup]: When flipping the

pschuh committed 2 years ago

e637bb39

Older