FlaxVision
Apache-2.0Python
A collection of neural network models ported from torchvision for use with JAX and Flax.
46 stars2 forks0 contributors
Overview
A selection of neural network models ported from torchvision for JAX & Flax.
Related Projects
Community-curated · Updated weekly · 100% open source
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
