Vit Python Packages

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.

inference-gpu

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.

inference-cli

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.

inference-core

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

nnm

Neural Network Models

towhee

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

jabberjay

🦜 Synthetic Voice Detection

morphottention

Mathematical Morphology-based self-attention module for PyTorch (CUDA) using Flash-style kernel fusion.

towhee-models

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

dinotool

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

mnistvit

PyTorch implementation of a vision transformer for training on MNIST

clipq

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

smart-reid

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

navit-torch

My implementation of "Patch n’ Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution"

vision-llama

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

autodistill-vit

ViT module for use with autodistill.

passl

PASSL包含 SimCLR，MoCo v1/v2，BYOL，CLIP，PixPro，simsiam, SwAV, BEiT，MAE 等图像自监督算法以及 Vision Transformer，DEiT，Swin Transformer，CvT，T2T-ViT，MLP-Mixer，XCiT，ConvNeXt，PVTv2 等基础视觉算法

vit-rgts

Open source implementation of "Vision Transformers Need Registers"

mcvit

Implementation of the model: "(MC-ViT)" from the paper: "Memory Consolidation Enables Long-Context Video Understanding"

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.