A collection of neural network models ported from torchvision for use with JAX and Flax.
A selection of neural network models ported from torchvision for JAX & Flax.
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.