rwightman

Ross Wightman's starred repositories

stable-diffusion

A latent text-to-image diffusion model

Language:Jupyter NotebookNOASSERTION66006 555 696

gradio

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

Language:PythonApache-2.029868 160 4265

dalle-mini

DALL·E Mini - Generate images from a text prompt

Language:PythonApache-2.014673 111 155

DALLE2-pytorch

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

Language:PythonMIT10905 122 207

open_clip

An open source implementation of CLIP.

Language:Jupyter NotebookNOASSERTION8790 76 439

mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

Language:PythonNOASSERTION6869 58 184

ConvNeXt

Code release for ConvNeXt model

Language:PythonMIT5594 33 130

img2dataset

Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.

Language:PythonMIT3353 30 249

clip-retrieval

Easily compute clip embeddings and build a clip retrieval system with them

Language:Jupyter NotebookMIT2197 23 221

s4

Structured state space sequence models

Language:Jupyter NotebookApache-2.02183 49 128

fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.

Language:PythonApache-2.01911 41 77

big_vision

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

Language:Jupyter NotebookApache-2.01854 34 42

hivemind

Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.

Language:PythonMIT1848 57 160

ml-cvnets

CVNets: A library for training computer vision networks

Language:PythonNOASSERTION1703 33 91

EfficientFormer

EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]

Language:PythonNOASSERTION958 36 57

SLIP

Code release for SLIP Self-supervision meets Language-Image Pre-training

Language:PythonMIT730 17 27

x-clip

A concise but complete implementation of CLIP with various experimental improvements from recent papers

Language:PythonMIT660 25 14

wise-ft

Robust fine-tuning of zero-shot models

Language:PythonNOASSERTION579 6 23

ResizeRight

The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).

Language:PythonMIT541 12 13

CLIP_benchmark

CLIP-like model evaluation

Language:Jupyter NotebookMIT513 12 64

aesthetic-predictor

A linear estimator on top of clip to predict the aesthetic quality of pictures

Language:Jupyter NotebookMIT397 12 6

EdgeNeXt

[CADL'22, ECCVW] Official repository of paper titled "EdgeNeXt: Efficiently Amalgamated CNN-Transformer Architecture for Mobile Vision Applications".

Language:PythonMIT329 6 20

GradCache

Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

Language:PythonApache-2.0317 9 25

examples

Example code and applications for machine learning on Graphcore IPUs

Language:PythonMIT313 44 3

PASS

The PASS dataset: pretrained models and how to get the data

Language:PythonMIT261 7 8

poptorch

PyTorch interface for the IPU

Language:C++MIT173 16 18

vae-textures

Texture mapping with variational auto-encoders

Language:Python40 40

detectron2_timm

A simple wrapper library for binding timm models as detectron2 backbones

Language:Jupyter Notebook38 2 1

open_clip_juwels

An open source implementation of CLIP.

Language:PythonNOASSERTION31 20

clip_benchmark

clip retrieval benchmark

Language:PythonMIT17 1 4