google/flax

Pull Requests Commits

Merge pull request #3600 from google:nnx-clean-flaglib

a-googler committed 1 year ago

cb6843f2

Merge pull request #3581 from google:nnx-submodule-iterator

a-googler committed 1 year ago

3b23351e

delete flaglib duplicated copyright comment

cgarciae committed 1 year ago

039845ce

[nnx] add submodule iterator

cgarciae committed 1 year ago

1e75509a

Merge pull request #3591 from shoyer:patch-1

a-googler committed 1 year ago

85eb8c05

Merge pull request #3582 from google:nnx-flagslib-add-get-overloads

a-googler committed 1 year ago

cff22d3f

Merge pull request #3583 from google:nnx-add-missing-symbols

a-googler committed 1 year ago

bfe04e56

Merge pull request #3596 from google:nnx-why-open-in-colab

a-googler committed 1 year ago

b28c0f74

Merge pull request #3578 from google:nnx-fix-variable-unflatten

a-googler committed 1 year ago

2d15e85f

Merge pull request #3592 from google:nnx-improve-tuple-set

a-googler committed 1 year ago

4fb56043

Merge pull request #3580 from google:nnx-get-named-sharding

a-googler committed 1 year ago

e8a14af9

Merge pull request #3587 from chiamp:error_format

a-googler committed 1 year ago

b48d05ff

add open in colab button

cgarciae committed 1 year ago

b6a096b0

improve graph_utils._set_key_tuple

cgarciae committed 1 year ago

3fcc0130

Fix type in NNX readme

shoyer committed 2 years ago

Verified 1c7ff8b7

fixed error doc formatting

chiamp committed 2 years ago

62859a6f

Merge pull request #3571 from giovannic:bug/attention_promotion

a-googler committed 2 years ago

85dfad24

[nnx] export missing symbols

cgarciae committed 2 years ago

03ffeac9

[nnx] flaglib add get overloads

cgarciae committed 2 years ago

2eebcc00

[nnx] improve spmd

cgarciae committed 2 years ago

3b772808

nnx: fix variable unflatten

cgarciae committed 2 years ago

0bf48f65

Merge pull request #3574 from chiamp:nnx_initializers

a-googler committed 2 years ago

008c3cc7

Merge pull request #3573 from google:nnx-improve-readme

a-googler committed 2 years ago

f1cb0114

expose ones and zeros initializers

chiamp committed 2 years ago

b5ec12b8

Merge pull request #3567 from google:nnx-add-dropout-test

a-googler committed 2 years ago

3e56f8e4

Merge pull request #3555 from chiamp:nnx_initializers

a-googler committed 2 years ago

dad384c3

Merge pull request #3527 from NeilGirdhar:key_type_error

a-googler committed 2 years ago

1f5947fc

nnx: add interacting with JAX section to README

cgarciae committed 2 years ago

0a67fc77

Fix promotion bug in MultiHeadDotProductAttention:

giovannic committed 2 years ago

1c27546e

Merge pull request #3561 from chiamp:typo

a-googler committed 2 years ago

f154a64d

Newer Older