google/flax

Pull Requests Commits

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

add Module.module_paths

Cristian Garcia committed 2 years ago

23952247

added norm equivalence tests

chiamp committed 2 years ago

9e921d08

updated nowrap docstring

chiamp committed 2 years ago

73586787

Merge pull request #3652 from chiamp:instancenorm

a-googler committed 2 years ago

61ece402

added InstanceNorm

chiamp committed 2 years ago

1ad60651

Copybara import of the project:

a-googler committed 2 years ago

f3c5199a

Replacing jax.tree_util.tree_map with mapping over leafs.

a-googler committed 2 years ago

fc777137

Add module_paths method to docs

cgarciae committed 2 years ago

Verified b6e950da

Merge pull request #3497 from chiamp:rng_guide

a-googler committed 2 years ago

a93b7a87

Merge pull request #3654 from google:module_paths

a-googler committed 2 years ago

0f0a65f5

make Sequential.__call__ compact

Cristian Garcia committed 2 years ago

3bc5c7cd

add Module.module_paths

cgarciae committed 2 years ago

a638b16f

Add explicit control over frozen/slots setting in flax.struct.dataclass

a-googler committed 2 years ago

3cd34b6b

Merge pull request #3643 from chiamp:partition

a-googler committed 2 years ago

70abeef8

Merge pull request #3641 from google:dependabot/pip/examples/vae/pillow-10.2.0

a-googler committed 2 years ago

1d234251

Merge pull request #3649 from chiamp:bump_version

a-googler committed 2 years ago

c1b2c13d

Older