https://github.com/rolandgvc/flaxvision
Python44
4 years ago
A selection of neural network models ported from torchvision for JAX & Flax.
Apache License 2.0