flaxvision

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

Search similar apps

    License

    Apache License 2.0

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

    Creator

    rolandgvc

    Related apps