Eqxvision
A Python package providing popular computer vision model architectures built with Equinox for JAX.
What is Eqxvision?
Eqxvision is a Python package that provides implementations of popular computer vision model architectures using the Equinox library for JAX. It solves the problem of accessing well-known vision models like AlexNet, FCN, and DeepLabV3 within the JAX ecosystem, enabling researchers and developers to leverage JAX's performance benefits for vision tasks.
Machine learning researchers and developers working with JAX who need computer vision models for tasks like image classification and segmentation, particularly those transitioning from PyTorch who want familiar architectures.
Developers choose Eqxvision because it brings torchvision-like functionality to JAX with Equinox's clean API, offering pretrained models, JAX transformation compatibility, and performance optimizations not readily available in other JAX vision libraries.
Overview
A Python package of computer vision models for the Equinox ecosystem.
Use Cases
Best For
- Implementing computer vision models in JAX with Equinox
Related Projects
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.
