A selection of neural network models ported from torchvision for JAX & Flax.
updated at April 11, 2024, 2:03 p.m.
4 +0
44 +0
2 +0