google/flax

Pull Requests Commits

Add preceding zeros to checkpoint files so they appear sorted under a directory.

Mohammad Norouzi committed 4 years ago

d83f6377

Merge pull request #1661 from sgrigory:971-add-circular-padding

a-googler committed 4 years ago

df266804

Merge pull request #1574 from google:avital-prefetch-doc

a-googler committed 4 years ago

46126954

Merge branch 'main' into avital-prefetch-doc

marcvanzee committed 4 years ago

Verified becff1cf

Merge pull request #1416 from marcvanzee:rtd

a-googler committed 4 years ago

64997959

reformat comments to circular padding in ConvTranspose

sgrigory committed 4 years ago

c4e1250f

add comments to circular padding in ConvTranspose

sgrigory committed 4 years ago

95ea6652

add circular padding to transposed convolution

sgrigory committed 4 years ago

969073db

add circular padding to linen.Conv

sgrigory committed 4 years ago

d5c07565

Fix docstring module

marcvanzee committed 4 years ago

ab38a002

Args --> Attributes

marcvanzee committed 4 years ago

a5e139d1

marcvanzee committed 4 years ago

1dbd90ea

Merge pull request #1672 from google:marcvanzee-patch-13

a-googler committed 4 years ago

6da4a003

Clarifies ordering of dimensions in Convs

marcvanzee committed 4 years ago

Verified 0edd5189

Merge pull request #1668 from jheek/mission-statement

avital committed 4 years ago

Verified d6a21943

Apply code review comments

avital committed 4 years ago

Verified 8630d2be

Post mission statement

jheek committed 4 years ago

319fa095

Merge pull request #1649 from jheek:enable-named-call-docs

a-googler committed 4 years ago

a9bdc363

Merge pull request #1657 from lkhphuc:patch-1

a-googler committed 4 years ago

390c7cfb

Merge pull request #1626 from andsteing:doc2

a-googler committed 4 years ago

2eb9df92

Fix GroupNorm type hint.

lkhphuc committed 4 years ago

fdf66c0e

Add named_call overrides to docs

jheek committed 4 years ago

f977dee1

Merge pull request #1570 from isaaccorley:activations/prelu

a-googler committed 4 years ago

004a118b

Merge pull request #1647 from google:marcvanzee-patch-13

a-googler committed 4 years ago

4cbf6bfa

Add comment to SST2

marcvanzee committed 4 years ago

Verified eff30d44

Test refactoring

jheek committed 4 years ago

33f7ead0

updated per suggestions

isaaccorley committed 4 years ago

73b9e0ca

isaaccorley committed 4 years ago

7b074932

update per suggestions

isaaccorley committed 4 years ago

40a54b9d

add linen activation tests

isaaccorley committed 4 years ago

4da0d98a

Older