google/flax

Pull Requests Commits

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

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

Raise an error for a bad key type

NeilGirdhar committed 2 years ago

23a0f918

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

Newer Older