google/flax

Pull Requests Commits

Merge pull request #3675 from chiamp:kwargs

a-googler committed 2 years ago

182afc02

Merge pull request #3663 from chiamp:mask

a-googler committed 2 years ago

0a88cfd2

added kwargs to self.param and self.variable

chiamp committed 2 years ago

bb86d523

Merge pull request #3685 from chiamp:rng_guide

a-googler committed 2 years ago

37fe9c93

Merge pull request #3684 from IvyZX:push2

a-googler committed 2 years ago

29f5264f

fixed rng guide outputs

chiamp committed 2 years ago

1e712cbe

add +1 to version after release

IvyZX committed 2 years ago

4a609961

Merge pull request #3682 from IvyZX:push

a-googler committed 2 years ago

598958ea

IvyZX committed 2 years ago

c2d752ec

Merge pull request #3646 from google:compact_name_scope_v3

a-googler committed 2 years ago

48f3a704

[nnx] add Module.partial_init

cgarciae committed 2 years ago

fae4d1f3

Merge pull request #3672 from chiamp:linen_intro

a-googler committed 2 years ago

d9585e0a

add compact_name_scope v3

cgarciae committed 2 years ago

b183dc6c

re-added linen_intro

chiamp committed 2 years ago

716cf49d

Merge pull request #3670 from chiamp:vmap

a-googler committed 2 years ago

6b70fe87

Merge pull request #3562 from chiamp:nnx_linear

a-googler committed 2 years ago

aee1aa10

updated vmap docstring

chiamp committed 2 years ago

40c2d94d

changed kwargs to actual key-word args

chiamp committed 2 years ago

4400343c

Merge pull request #3624 from chiamp:typing

a-googler committed 2 years ago

89954205

added flax.typing

chiamp committed 2 years ago

49ba1551

Merge pull request #3667 from chiamp:instancenorm

a-googler committed 2 years ago

82667a59

Merge pull request #3669 from chiamp:rng

a-googler committed 2 years ago

74bdc7b1

add default make_rng

chiamp committed 2 years ago

669815cf

Cristian Garcia committed 2 years ago

12b3db0c

Merge pull request #3657 from google:add-module_paths-to-docs

a-googler committed 2 years ago

cec9a46d

renamed channel_axes to feature_axes in InstanceNorm

chiamp committed 2 years ago

59d5c2ce

Merge pull request #3661 from chiamp:nowrap

a-googler committed 2 years ago

0a39a695

Merge pull request #3662 from chiamp:norm_test

a-googler committed 2 years ago

6e4a6429

enforce mask kwarg in norm layers

chiamp committed 2 years ago

1cf75930

add Module.module_paths

Cristian Garcia committed 2 years ago

23952247

Newer Older