google/flax

Pull Requests Commits

marcvanzee committed 4 years ago

9fada159

Merge pull request #1119 from avital:remove-deprecated-examples

a-googler committed 4 years ago

d047b823

Copybara import of the project:

andsteing committed 4 years ago

03d2b983

support batch dimension in image summaries

iansimon committed 4 years ago

6e0c0f46

Merge pull request #1112 from brettkoonce:wmt-tweak

a-googler committed 4 years ago

ae1d860b

Remove deprecated LM1b example

avital committed 4 years ago

b02a2062

Merge pull request #1113 from brettkoonce:mnist-readme-tweak

a-googler committed 4 years ago

15b6229d

marcvanzee committed 4 years ago

f04659c7

Add pathlib / fspath compatibility to SummaryWriter

Conchylicultor committed 4 years ago

afa4a2cd

mnist: tweak demo call

brettkoonce committed 4 years ago

4f0760cc

wmt: tweak readme example config

brettkoonce committed 4 years ago

91a937cb

Merge pull request #1100 from jheek:improve-bind-docs

a-googler committed 4 years ago

748f7a38

Merge pull request #1105 from sethvargo/patch-1

marcvanzee committed 4 years ago

Verified fde4b1e1

Improve Actions hygiene

sethvargo committed 4 years ago

Verified e81d2b10

Merge pull request #1084 from marcvanzee:bugfix2

a-googler committed 4 years ago

3bd05010

Fixes a type in ensembling HOWTO

marcvanzee committed 4 years ago

Verified e6dc4ab2

improve bind docs

jheek committed 4 years ago

73715f11

Merge branch 'master' of https://github.com/google/flax into bugfix2

marcvanzee committed 4 years ago

e71c7358

Explicitly cast JAX PRNGKey seed to uint32

Jake VanderPlas committed 4 years ago

29d734e6

Merge pull request #1099 from jheek:add-pytreenode-docs

a-googler committed 4 years ago

3f13b23b

Show PyTreeNode on RTD

jheek committed 4 years ago

852cece1

This change is needed to allow using @flax.struct.dataclass on a class that inherits itself from another dataclass created with @flax.struct.dataclass.

andsteing committed 4 years ago

e53532f2

Don't turn clz into a dataclass if it's already a dataclass.

Marvin182 committed 4 years ago

6dfd7576

Fix the typo in the comment of `Encoder1DBlock`; correct "Args" with "Attributes"; correct import order; suppress `g-bare-generic` warning.

a-googler committed 4 years ago

e972aa85

Merge pull request #1088 from 8bitmp3:patch-1

a-googler committed 4 years ago

757b87ca

Merge pull request #1093 from andsteing:examples

a-googler committed 4 years ago

dd3b52fa

8bitmp3 committed 4 years ago

Verified 998b9128

Merge pull request #1078 from jheek:nn-apply

a-googler committed 4 years ago

d4cb43b0

Remove method arg from nn.apply

jheek committed 4 years ago

d55057b6

Update flax/core/scope.py

jheek committed 4 years ago

77a4b1ca

Older