flaxvision

A selection of neural network models ported from torchvision for JAX & Flax.

44 stars

4 watching

2 forks

Language: Python

last commit: over 3 years ago

Linked from 1 awesome list

Backlinks from these awesome lists:

n2cholas/awesome-jax