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
ALXGoogle Research
Stars38,119
Forks8,427
Last commit1 day ago
OryxProbabilistic reasoning and statistical analysis in TensorFlow
Stars4,418
Forks1,122
Last commit1 month ago
BRAXMassively parallel rigidbody physics simulation on accelerator hardware.
Stars3,191
Forks343
Last commit25 days ago
MctxMonte Carlo tree search in JAX
Stars2,631
Forks209
Last commit2 days ago
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.