rwightman

nanoGPT

The simplest, fastest repository for training/finetuning medium-sized GPTs.

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

StableLM

StableLM: Stability AI Language Models

surya

OCR, layout analysis, reading order, line detection in 90+ languages

open_flamingo

An open-source framework for training large multimodal models.

Otter

🦦 Otter, a multi-modal model based on OpenFlamingo (open-sourced version of DeepMind's Flamingo), trained on MIMIC-IT and showcasing improved instruction-following and in-context learning ability.

doctr

docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning.

musiclm-pytorch

Implementation of MusicLM, Google's new SOTA model for music generation using attention networks, in Pytorch

EVA

EVA Series: Visual Representation Fantasies from BAAI

GLIP

Grounded Language-Image Pre-training

huggingface_hub

The official Python client for the Huggingface Hub.

datatrove

Freeing data processing from scripting madness by providing a set of platform-agnostic customizable pipeline processing blocks.

ConvNeXt-V2

Code release for ConvNeXt V2 model

coyo-dataset

COYO-700M: Large-scale Image-Text Pair Dataset

torchdynamo

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

phenaki-pytorch

Implementation of Phenaki Video, which uses Mask GIT to produce text guided videos of up to 2 minutes in length, in Pytorch

tensordict

TensorDict is a pytorch dedicated tensor container.

datacomp

DataComp: In search of the next generation of multimodal datasets

pix2struct

maxvit

[ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmentation, image quality, and generative modeling...

robotic-transformer-pytorch

Implementation of RT1 (Robotic Transformer) in Pytorch

cc2dataset

Easily convert common crawl to a dataset of caption and document. Image/text Audio/text Video/text, ...

pypdfium2

Python bindings to PDFium

mlx-llm

Large Language Models (LLMs) applications and tools running on Apple Silicon in real-time with Apple MLX.

Battle-of-the-Backbones

chug

Minimal sharded dataset loaders, decoders, and utils for multi-modal document, image, and text datasets.

apted

Python APTED algorithm for the Tree Edit Distance

timm-models-explorer

Timm model explorer

posenet-pytorch

A PyTorch port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)

timm-lr-scheduler-explorer

A dashboard for exploring timm learning rate schedulers