flax
c53c1c53 - Merge pull request #1312 from JamesAllingham/add-bn-vmap-to-state-howto

Commit

4 years ago

Merge pull request #1312 from JamesAllingham/add-bn-vmap-to-state-howto Add vmap + BatchNorm example to the state howto

References

Author

jheek

Parents