flax
Add vmap + BatchNorm example to the state howto
#1312

Merged

Add vmap + BatchNorm example to the state howto #1312

jheek merged 1 commit into google:master from JamesAllingham:add-bn-vmap-to-state-howto

Add vmap + BatchNorm example to the state howto

00e9fea7

google-cla added cla: no

google-cla removed cla: no

google-cla added cla: yes

jheek approved these changes on 2021-05-17

jheek added pull ready

jheek merged c53c1c53 into master 4 years ago

Reviewers

jheek

Assignees

No one assigned

Labels

cla: yes pull ready

Milestone

No milestone