google/flax

Pull Requests Commits

Make static type checkers aware the fact that all derived classes of `FiddleBaseParameterizable` have a dataclass-like semantics (ie having fields).

a-googler committed 2 years ago

0b977337

Make the injectable `dot_general` optional and not pre-initialized so that method interceptor works.

IvyZX committed 2 years ago

fac0cef3

Remove import checks that are no longer valid.

IvyZX committed 2 years ago

3db72f59

Merge pull request #3412 from chiamp:changelog

a-googler committed 2 years ago

342d12ab

added attention refactor to changelog

chiamp committed 2 years ago

5892e2bc

Merge pull request #3405 from chiamp:weightnorm

a-googler committed 2 years ago

d0d04390

added weightnorm layer

chiamp committed 2 years ago

59054a29

Merge pull request #3408 from chiamp:toml

a-googler committed 2 years ago

008880fc

chiamp committed 2 years ago

1a0961e4

Copybara import of the project:

chiamp committed 2 years ago

b76f4878

Merge pull request #3370 from chiamp:docs

a-googler committed 2 years ago

5a1fbc20

Merge pull request #3406 from chiamp:rmsnorm

a-googler committed 2 years ago

816cc5e9

Merge pull request #3267 from 8bitmp3:flax-faq

a-googler committed 2 years ago

30147a4c

added rmsnorm to api docs

chiamp committed 2 years ago

35326d48

Update Flax F.A.Q.

8bitmp3 committed 2 years ago

7b1d8a61

Add Flax F.A.Q.

8bitmp3 committed 2 years ago

d0216d31

Merge pull request #3383 from chiamp:imagenet

a-googler committed 2 years ago

66354365

Train/eval imagenet fake data benchmark for a single step.

andsteing committed 2 years ago

97e2fea7

Merge pull request #3404 from chiamp:filter_warning

a-googler committed 2 years ago

37f9bf13

chiamp committed 2 years ago

27852f6a

Merge pull request #3401 from levskaya:pr3396

a-googler committed 2 years ago

6275ff19

levskaya committed 2 years ago

9521cf99

Update `kernel_size` docstring for `nn.Conv{local}`.

tomwardio committed 2 years ago

0347dd2f

Updates Colab runtime version Colabs.

andsteing committed 2 years ago

55e12904

Merge pull request #3394 from google:test-ci-3

a-googler committed 2 years ago

f20aed42

constrain numpy and clu

cgarciae committed 2 years ago

d34f60c4

add truncated_normal to initializers

levskaya committed 2 years ago

Verified 6f4ef6ab

updated imagenet readme

chiamp committed 2 years ago

83881040

Fix PPO example build

romanngg committed 2 years ago

d059ba8a

Add FLOPs calculations to getting started example

romanngg committed 2 years ago

54a42daf

Older