flax
00e9fea7 - Add vmap + BatchNorm example to the state howto

Commit

4 years ago

Add vmap + BatchNorm example to the state howto

References

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

Author

JamesAllingham

JamesAllingham

Parents

Loading