google/flax

Pull Requests Commits

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

a-googler committed 2 years ago

b4682076

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 features arg

chiamp committed 2 years ago

aa1703b9

Merge pull request #3553 from chiamp:format

a-googler committed 2 years ago

44a8657b

Merge pull request #3550 from chiamp:attention

a-googler committed 2 years ago

2414d1cc

format linen_linear_test.py

chiamp committed 2 years ago

71b5c464

changed to for attention layer

chiamp committed 2 years ago

7f10ad8d

Merge pull request #3548 from chiamp:nnx_attention

a-googler committed 2 years ago

2e51a4e5

Merge pull request #3537 from chiamp:nnx_conv

a-googler committed 2 years ago

73a0562e

added sow attention weights to NNX

chiamp committed 2 years ago

403ab153

add nnx conv support for int kernel size

chiamp committed 2 years ago

6c5f44a8

Merge pull request #3545 from chiamp:backquote

a-googler committed 2 years ago

51e2cf53

fixed double backquote code font

chiamp committed 2 years ago

b1f6aa21

Merge pull request #3544 from chiamp:module_docs

a-googler committed 2 years ago

0c213228

Merge pull request #3542 from google:linen-fix-scan-out-axes-p3

a-googler committed 2 years ago

ee105de1

add module methods to api docs

chiamp committed 2 years ago

8424b9b5

Merge pull request #3539 from chiamp:embed

a-googler committed 2 years ago

e29b1b36

updated embed docstring

chiamp committed 2 years ago

2b3ca282

add test_scan_negative_axes

cgarciae committed 2 years ago

a47eb363

Merge pull request #3540 from google:linen-fix-scan-out-axes

a-googler committed 2 years ago

870847ed

fix scan out_axes

cgarciae committed 2 years ago

9aa8ec68

Merge pull request #3529 from chiamp:attention

a-googler committed 2 years ago

50cd169d

Ensure that `_hashable_filter` does not convert strings to a tuple of letters

superbobry committed 2 years ago

3687af03

added sow attention weights

chiamp committed 2 years ago

9d989b0b

Merge pull request #3526 from Micky774:nnx_conv_test_trim

a-googler committed 2 years ago

829e004c

Merge pull request #3531 from chiamp:normalize

a-googler committed 2 years ago

512a6d88

Older