google/flax

Pull Requests Commits

[nnx] port lm1b

cgarciae committed 1 year ago

726cbc7f

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

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

bump version number

chiamp committed 2 years ago

ff313359

Merge pull request #3644 from chiamp:release

a-googler committed 2 years ago

2a84d42d

chiamp committed 2 years ago

a9c54997

fixed docstring

chiamp committed 2 years ago

ccf893f0

add compact_name_scope v2

a-googler committed 2 years ago

3b1a8073

Bump pillow from 10.0.1 to 10.2.0 in /examples/vae

dependabot[bot] committed 2 years ago

Verified 8f5c61fa

add compact_name_scope v2

Cristian Garcia committed 2 years ago

97e9c704

Merge pull request #3634 from chiamp:batch_apply

a-googler committed 2 years ago

77da09e9

Remove calls to deprecated function jax.lax.tie_in

Jake VanderPlas committed 2 years ago

993aa2c5

Copybara import of the project:

chiamp committed 2 years ago

0370a610

Copybara import of the project:

a-googler committed 2 years ago

4d075e53

Merge pull request #3621 from google:nontransparent

a-googler committed 2 years ago

7e9bb761

added BatchApply

chiamp committed 2 years ago

6454502a

added rng_guide

chiamp committed 2 years ago

fdf9baad

add compact_name_scope

cgarciae committed 2 years ago

c9e63a7b

support transform reuse for lift.jit decorator

Cristian Garcia committed 2 years ago

93d3eb78

Merge pull request #3625 from chiamp:ruff

a-googler committed 2 years ago

c6c169a3

Older