google/flax

Pull Requests Commits

quantile regression

cgarciae committed 2 years ago

3fb1d12f

[nnx] port lm1b

cgarciae committed 2 years ago

a77b01a9

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

Merge pull request #3561 from chiamp:typo

a-googler committed 2 years ago

f154a64d

Merge pull request #3563 from chiamp:readme

a-googler committed 2 years ago

ca7d7483

Merge pull request #3566 from google:nnx-attention-add-qkv-features

a-googler committed 2 years ago

a59675c1

Merge pull request #3547 from google:nnx-rngs-impl-getattr

a-googler committed 2 years ago

eb1bfb11

Passed previously unused `iterate_clz` to `register_pytree_with_keys`.

superbobry committed 2 years ago

0697987e

Merge pull request #3565 from google:cgarciae-nnx-fix-readme-install

a-googler committed 2 years ago

fa2d80ef

add kwargs support for nn.jit

Cristian Garcia committed 2 years ago

cf7311da

nnx: add complex test with scan + batchnorm + dropout

cgarciae committed 2 years ago

df60f611

implement Rng.__getattr__

cgarciae committed 2 years ago

8520ad94

nnx: add qkv_features back to MHA

cgarciae committed 2 years ago

eb850d87

Merge pull request #3523 from google:improve-variables-mapping

a-googler committed 2 years ago

b4682076

[nnx] Fix readme install instruction

cgarciae committed 2 years ago

Verified be01a263

chiamp committed 2 years ago

d7eb6cdd

chiamp committed 2 years ago

64fffb01

Merge pull request #3560 from google:cgarciae-patch-1

a-googler committed 2 years ago

c9970933

Merge pull request #3556 from chiamp:readme

a-googler committed 2 years ago

0a337cbc

Merge pull request #3554 from chiamp:nnx_features

a-googler committed 2 years ago

82e1a27d

Disable ruff sort imports

cgarciae committed 2 years ago

Verified 4e852f57

updated NNX readme

chiamp committed 2 years ago

49e6f4f1

re-factored nnx initializers

chiamp committed 2 years ago

1b7493c0

re-factored features arg

chiamp committed 2 years ago

aa1703b9

Older