Starred repositories
Use late-interaction multi-modal models such as ColPali in just a few lines of code.
[CVPR 2024 Highlight] Mitigating Object Hallucinations in Large Vision-Language Models through Visual Contrastive Decoding
Efficient Triton Kernels for LLM Training
Helpful tools and examples for working with flex-attention
This is a Phi-3 book for getting started with Phi-3. Phi-3, a family of open AI models developed by Microsoft. Phi-3 models are the most capable and cost-effective small language models (SLMs) avai…
The fastest way to create an HTML app
High-quality datasets, tools, and concepts for LLM fine-tuning.
Easily use and train state of the art late-interaction retrieval methods (ColBERT) in any RAG pipeline. Designed for modularity and ease-of-use, backed by research.
A pytorch quantization backend for optimum
Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.
Make drawing and labeling bounding boxes easy as cake
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools
We write your reusable computer vision tools. 💜
Strong and Open Vision Language Assistant for Mobile Devices
the AI-native open-source embedding database
Official PyTorch implementation of "TinySAM: Pushing the Envelope for Efficient Segment Anything Model"
Finetune Llama 3.1, Mistral, Phi & Gemma LLMs 2-5x faster with 80% less memory
An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algorithm.
A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API
You like pytorch? You like micrograd? You love tinygrad! ❤️
A guidance language for controlling large language models.
Alpaca dataset from Stanford, cleaned and curated
A scikit-learn compatible neural network library that wraps PyTorch
It's a cooler way to store simple linear models.
The pytest framework makes it easy to write small tests, yet scales to support complex functional testing