google/flax

Pull Requests Commits

Conchylicultor committed 4 years ago

b6743fa8

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

implement functional init/apply and bind.

jheek committed 4 years ago

05aab84e

Creates logdir before logging to files.

andsteing committed 4 years ago

57d63373

Improve Flax Linen API __call__ and __init__ formatting in docstrings

8bitmp3 committed 4 years ago

Verified 054a3075

Fixes a bug in Module

marcvanzee committed 4 years ago

b0c1e2fa

Merge pull request #1083 from google:revert-1075-fix-error

a-googler committed 4 years ago

917f4470

Revert "Fixes a bug in Module and adds test"

marcvanzee committed 4 years ago

Verified 4e8f02aa

Merge pull request #1075 from marcvanzee:fix-error

a-googler committed 4 years ago

816b5ba0

Merge pull request #1070 from marcvanzee:errors

a-googler committed 4 years ago

34ab6da3

Fixes a bug in Module and adds test

marcvanzee committed 4 years ago

73a23415

Merge pull request #1082 from google/marcvanzee-patch-7

marcvanzee committed 4 years ago

Verified d82de14a

Older