PR #2831 Improve nn.map_variables example

Improve nn.map_variables example #2831

copybara-service merged 2 commits into google:main from cgarciae:improve-map_variables-v2

cgarciae force pushed 3 years ago

cgarciae requested a review from

chiamp 3 years ago

cgarciae assigned

cgarciae 3 years ago

cgarciae changed the title ~~Improve nn.map_variable example~~ Improve nn.map_variables example 3 years ago

improve the CasualDense example

93e5254e

cgarciae force pushed to 93e5254e 3 years ago

chiamp commented on 2023-02-01

Update flax/linen/transforms.py

5f67729d

chiamp added pull ready

copybara-service merged f2094bac into main 3 years ago

Reviewers

chiamp

Assignees

cgarciae

Labels

pull ready

Milestone

No milestone