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,711
Forks8,465
Last commit20 hours ago
OryxProbabilistic reasoning and statistical analysis in TensorFlow
Stars4,432
Forks1,127
Last commit1 month ago
BRAXMassively parallel rigidbody physics simulation on accelerator hardware.
Stars3,230
Forks349
Last commit1 month ago
MctxMonte Carlo tree search in JAX
Stars2,657
Forks216
Last commit1 month 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.