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-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.

morphottention

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

towhee

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

towhee-models

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

clipq

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

mnistvit

A PyTorch implementation of a vision transformer for training on MNIST

jabberjay

🦜 Synthetic Voice Detection

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.

vit-rgts

Open source implementation of "Vision Transformers Need Registers"

transformerx

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

vit-trainer

Explore fine-tuning the Vision Transformer (ViT) model for object recognition in robotics using PyTorch. This tutorial covers setup, training, and evaluation processes, achieving impressive accuracy with practical resource constraints. Ideal for learners in AI and robotics.

smart-reid

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

mcvit

Paper - Pytorch