google/flax

Pull Requests Commits

added geglu activation and tests

HenryNdubuaku committed 2 years ago

c5ca7d71

Add NNX/Linen API consistency test for `Conv` layer

Micky774 committed 2 years ago

0daa8dfb

Merge pull request #3498 from Micky774:prelu_correctness_check

a-googler committed 2 years ago

c3c4cbf7

simplify MultiHeadAttention API

cgarciae committed 2 years ago

0a1f78a3

Fix types in example code.

DrMarcII committed 2 years ago

90158f4c

Updated tests to avoid deprecated `JaxTestCase`

Micky774 committed 2 years ago

eca581ac

add LinearGeneral and MultiHeadAttention

cgarciae committed 2 years ago

7ff6e4b6

Merge pull request #3502 from google:nnx-nested-state

a-googler committed 2 years ago

e172c768

Merge pull request #3503 from google:nnx-fix-why

a-googler committed 2 years ago

97b61674

allow integer access

cgarciae committed 2 years ago

7f8c6006

test State get/setattr

cgarciae committed 2 years ago

096adc85

switch to nested State representation

cgarciae committed 2 years ago

a8340bc6

add missing import

cgarciae committed 2 years ago

43087460

Improved rigor of PReLU test

Micky774 committed 2 years ago

8845a65d

No public description

DrMarcII committed 2 years ago

70214f4e

Merge pull request #3489 from chiamp:attention

a-googler committed 2 years ago

c41d4b47

removed selfattention from doctest

chiamp committed 2 years ago

417d3f54

Merge pull request #3462 from zhaoyang-0204:my_branch

a-googler committed 2 years ago

78b0c8cd

Merge pull request #3485 from chiamp:enum

a-googler committed 2 years ago

79d925fe

Merge pull request #3477 from chiamp:zsh

a-googler committed 2 years ago

e8203257

Merge pull request #3481 from chiamp:rst

a-googler committed 2 years ago

3c37eedd

Merge pull request #3482 from chiamp:ruff

a-googler committed 2 years ago

646da5b6

Merge pull request #3470 from chiamp:selfattention

a-googler committed 2 years ago

4a011d5e

Merge pull request #3486 from google:nnx-graph-utils

a-googler committed 2 years ago

47f763dc

add support for python container types

cgarciae committed 2 years ago

eef80668

added enum support for tabulate

chiamp committed 2 years ago

01049c71

Merge pull request #3450 from VictorPrins:loading-data-guide

a-googler committed 2 years ago

091df13e

add dataset loading guide to docs

VictorPrins committed 2 years ago

ab953319

adding doctest to .rst files

chiamp committed 2 years ago

81213ec9

disabled ruff formatter

chiamp committed 2 years ago

86fa50c1

Newer Older