google/flax

Pull Requests Commits

added init_cache docstring example

chiamp committed 1 year ago

c257e9e1

Merge pull request #3675 from chiamp:kwargs

a-googler committed 1 year ago

182afc02

Merge pull request #3663 from chiamp:mask

a-googler committed 1 year ago

0a88cfd2

added kwargs to self.param and self.variable

chiamp committed 1 year ago

bb86d523

Merge pull request #3685 from chiamp:rng_guide

a-googler committed 1 year ago

37fe9c93

Merge pull request #3684 from IvyZX:push2

a-googler committed 1 year ago

29f5264f

fixed rng guide outputs

chiamp committed 1 year ago

1e712cbe

add +1 to version after release

IvyZX committed 1 year ago

4a609961

Merge pull request #3682 from IvyZX:push

a-googler committed 1 year ago

598958ea

IvyZX committed 1 year ago

c2d752ec

Merge pull request #3646 from google:compact_name_scope_v3

a-googler committed 1 year ago

48f3a704

[nnx] add Module.partial_init

cgarciae committed 1 year ago

fae4d1f3

Merge pull request #3672 from chiamp:linen_intro

a-googler committed 1 year ago

d9585e0a

add compact_name_scope v3

cgarciae committed 1 year ago

b183dc6c

re-added linen_intro

chiamp committed 1 year ago

716cf49d

Merge pull request #3670 from chiamp:vmap

a-googler committed 1 year ago

6b70fe87

Merge pull request #3562 from chiamp:nnx_linear

a-googler committed 1 year ago

aee1aa10

updated vmap docstring

chiamp committed 1 year ago

40c2d94d

changed kwargs to actual key-word args

chiamp committed 1 year ago

4400343c

Merge pull request #3624 from chiamp:typing

a-googler committed 1 year ago

89954205

added flax.typing

chiamp committed 1 year ago

49ba1551

Merge pull request #3667 from chiamp:instancenorm

a-googler committed 1 year ago

82667a59

Merge pull request #3669 from chiamp:rng

a-googler committed 1 year ago

74bdc7b1

add default make_rng

chiamp committed 1 year ago

669815cf

Cristian Garcia committed 1 year ago

12b3db0c

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

a-googler committed 1 year ago

cec9a46d

renamed channel_axes to feature_axes in InstanceNorm

chiamp committed 1 year ago

59d5c2ce

Merge pull request #3661 from chiamp:nowrap

a-googler committed 1 year ago

0a39a695

Merge pull request #3662 from chiamp:norm_test

a-googler committed 1 year ago

6e4a6429

enforce mask kwarg in norm layers

chiamp committed 1 year ago

1cf75930

Newer Older