Vit Python Packages

inference-gpu

Turn any computer or edge device into a command center for your computer vision projects.

inference-cli

Turn any computer or edge device into a command center for your computer vision projects.

inference-sdk

Turn any computer or edge device into a command center for your computer vision projects.

sageattention

[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across language, image, and video models.

inference

Turn any computer or edge device into a command center for your computer vision projects.

pix2tex

pix2tex: Using a ViT to convert images of equations into LaTeX code.

nnm

Neural Network Models

inference-core

Turn any computer or edge device into a command center for your computer vision projects.

inference-cpu

Turn any computer or edge device into a command center for your computer vision projects.

towhee

Towhee is a framework that is dedicated to making neural data processing pipelines simple and fast.

jabberjay

🦜 Synthetic Voice Detection

clipq

A simple implementation of a CLIP that splits up an image into quandrants and then gets the embeddings for each quandrant

vision-llama

Implementation of VisionLLaMA from the paper: "VisionLLaMA: A Unified LLaMA Interface for Vision Tasks" in PyTorch and Zeta

towhee-models

Towhee is a framework that helps you encode your unstructured data into embeddings.

navit-torch

navit - Pytorch

mnistvit

A PyTorch implementation of a vision transformer for training on MNIST

dinotool

Command-line tool for extracting DINOv3, CLIP, SigLIP2, RADIO, features for images and videos

autodistill-vit

ViT module for use with autodistill.

vit-rgts

vit-registers - Pytorch

smart-reid

Turn any computer or edge device into a command center for your computer vision projects.

transformerx

Flexible Python library providing building blocks (layers) for reproducible Transformers research (Tensorflow ✅, Pytorch 🔜, and Jax 🔜)

vit-trainer

A simple, educational package for fine-tuning Vision Transformers

mcvit

Paper - Pytorch

omegavit

OmegaViT (ΩViT) is a cutting-edge vision transformer architecture that combines multi-query attention, rotary embeddings, state space modeling, and mixture of experts to achieve superior performance across various computer vision tasks.