Ross Wightman

Author of timm; CV at Hugging Face

Authored Projects (1)

Starred by

Patrick von Platen

Patrick von Platen(Author of Hugging Face Diffusers; Research Engineer at Mistral).

gen-efficientnet-pytorch by rwightman

PyTorch image models for efficient architectures

Implements EfficientNet, MobileNetV3, MixNet, MNASNet, FBNet, Single-Path NAS.
Includes pre-trained weights ported from TensorFlow.
Supports ONNX and Caffe2 export/validation.
Offers JIT optimized Swish/Mish activation functions.

Created 6 years ago

Updated 1 year ago

Starred Projects (56)

gpt-oss by openai

Open-weight LLMs for reasoning and agents

karpathy:

borzunov:

chiphuyen:

transitive-bullshit:

Created 2 months ago

Updated 3 days ago

Janus by deepseek-ai

Unified multimodal model research paper for understanding and generation

jiamings:

omarsar:

taranjeet:

zhyncs:

Created 11 months ago

Updated 7 months ago

crawlee by apify

Web scraping/browser automation library for building reliable crawlers

wsxiaoys:

ValentaTomas:

transitive-bullshit:

Created 9 years ago

Updated 1 day ago

xlstm by NX-AI

Recurrent neural network architecture based on the original LSTM

jiamings:

luiscape:

osanseviero:

Created 1 year ago

Updated 2 weeks ago

C-Optim by kyleliang919

Improving transformer training with a single line of code

Created 1 year ago

Updated 3 weeks ago

pytorch-grad-cam by jacobgil

PyTorch SDK for explainable AI in computer vision

hammer:

rasbt:

Jiayi-Pan:

JohannesHa:

Created 8 years ago

Updated 5 months ago

transfusion-pytorch by lucidrains

Pytorch implementation for multimodal model research

lucidrains:

Created 1 year ago

Updated 3 months ago

cambrian by cambrian-mllm

Multimodal LLM research paper with vision-centric design

hammer:

infwinston:

hiyouga:

rstojnic:

Created 1 year ago

Updated 10 months ago

torchtitan by pytorch

PyTorch platform for generative AI model training research

karpathy:

lewtun:

zhuohan123:

CodeCreator:

Created 1 year ago

Updated 19 hours ago

datatrove by huggingface

Data processing library for large-scale text data

lewtun:

shizhediao:

apsdehal:

hammer:

Created 2 years ago

Updated 2 days ago

mlx-llm by riccardomusmeci

LLM tools/apps for Apple Silicon using MLX

awni:

osanseviero:

Created 1 year ago

Updated 7 months ago

Otter by EvolvingLMMs-Lab

Multimodal model for improved instruction following and in-context learning

jiamings:

chiphuyen:

parasj:

apsdehal:

Created 2 years ago

Updated 1 year ago

datacomp by mlfoundations

Competition tooling for multimodal dataset curation to pre-train CLIP models

shizhediao:

hammer:

chiphuyen:

osanseviero:

Created 2 years ago

Updated 4 months ago

StableLM by Stability-AI

Language models by Stability AI

geohot:

vincentweisser:

casper-hansen:

calvinfo:

Created 2 years ago

Updated 1 year ago

open_flamingo by mlfoundations

Open-source framework for training large multimodal models

chiphuyen:

winglian:

osanseviero:

hammer:

Created 2 years ago

Updated 1 year ago

nanoGPT by karpathy

Minimalist repo for training/finetuning GPT models

tobi:

danielgross:

chiphuyen:

ankane:

Created 2 years ago

Updated 9 months ago

robotic-transformer-pytorch by lucidrains

PyTorch implementation of RT1 (Robotic Transformer) for robotic control

codekansas:

Created 2 years ago

Updated 11 months ago

coyo-dataset by kakaobrain

Image-text pair dataset for vision-language model training

jn2clark:

ebursztein:

ablattmann:

JustinLin610:

Created 3 years ago

Updated 2 years ago

diffusers by huggingface

PyTorch/Flax library for diffusion model research and applications

karpathy:

clmnt:

vincentweisser:

hammer:

Created 3 years ago

Updated 12 hours ago

stable-diffusion by CompVis

Latent text-to-image diffusion model

gaearon:

patrickvonplaten:

codekansas:

ValentaTomas:

Created 3 years ago

Updated 1 year ago

EdgeNeXt by mmaaz60

Vision architecture research paper:

Created 3 years ago

Updated 2 years ago

wise-ft by mlfoundations

Research paper code for robust fine-tuning of zero-shot models

jn2clark:

Created 4 years ago

Updated 3 years ago

dalle-mini by borisdayma

Text-to-image model for generating images from text prompts

karpathy:

hardmaru:

infwinston:

codekansas:

Created 4 years ago

Updated 1 year ago

CLIP_benchmark by LAION-AI

CLI tool for CLIP-like model evaluation across diverse tasks/datasets

jn2clark:

patrickvonplaten:

Created 3 years ago

Updated 1 month ago

clip-retrieval by rom1504

CLIP retrieval system for semantic search

jeresig:

chenlin9:

yoland68:

simonw:

Created 4 years ago

Updated 1 month ago

DALLE2-pytorch by lucidrains

PyTorch implementation of DALL-E 2, a text-to-image synthesis neural network

lucidrains:

codekansas:

victortaelin:

jn2clark:

Created 3 years ago

Updated 1 year ago

open_clip by mlfoundations

OpenCLIP: open-source CLIP implementation for vision-language representation learning

ekzhang:

jeresig:

c4pt0r:

taranjeet:

Created 4 years ago

Updated 1 week ago

GradCache by luyug

Technique for contrastive learning beyond GPU memory limits

jph00:

Created 4 years ago

Updated 1 year ago

SLIP by facebookresearch

Research paper code for self-supervised language-image pre-training

chenlin9:

s9xie:

Created 3 years ago

Updated 2 years ago

examples by graphcore

ML examples for Graphcore IPUs, training and inference

pgarbacki:

parasj:

Created 7 years ago

Updated 1 year ago

img2dataset by rom1504

CLI tool for creating large image datasets from URLs

winglian:

aangelopoulos:

lucidrains:

jn2clark:

Created 4 years ago

Updated 6 days ago

mesh-transformer-jax by kingoflolz

JAX library for model-parallel transformers

gakonst:

parano:

nirga:

vincentweisser:

Created 4 years ago

Updated 2 years ago

unilm by microsoft

Foundation models for language, vision, speech, and multimodal tasks

jph00:

AlexCheema:

chiphuyen:

osanseviero:

Created 6 years ago

Updated 2 months ago

Transformers-Tutorials by NielsRogge

Transformer demos using Hugging Face, implemented in PyTorch

ogabrielluiz:

patrickvonplaten:

lantiga:

omarsar:

Created 5 years ago

Updated 2 months ago

decision-transformer by kzl

RL via sequence modeling research paper

aravindsrinivas:

hammer:

chenlin9:

jiamings:

Created 4 years ago

Updated 1 year ago

x-transformers by lucidrains

Transformer library with extensive experimental features

chiphuyen:

hammer:

chenlin9:

Jiayi-Pan:

Created 4 years ago

Updated 5 days ago

DALLE-pytorch by lucidrains

PyTorch implementation of DALL-E, OpenAI's text-to-image transformer

omarsar:

luiscape:

bfirsh:

chenlin9:

Created 4 years ago

Updated 1 year ago

awesome-jax by n2cholas

Curated list of JAX resources for high-performance ML research

aravindsrinivas:

hammer:

dynamicwebpaige:

patrickvonplaten:

Created 4 years ago

Updated 2 weeks ago

vision_transformer by google-research

Vision Transformer and MLP-Mixer models in JAX/Flax

aravindsrinivas:

gakonst:

jn2clark:

merrymercy:

Created 4 years ago

Updated 6 months ago

vit-pytorch by lucidrains

PyTorch library for Vision Transformer variants and related techniques

karpathy:

lucidrains:

chiphuyen:

forresti:

Created 5 years ago

Updated 1 month ago

flax by google

NN library for JAX, designed for flexibility in neural network research

charliermarsh:

codekansas:

Jiayi-Pan:

jaredpalmer:

Created 5 years ago

Updated 17 hours ago

minGPT by karpathy

Minimal PyTorch re-implementation for GPT training and inference

aravindsrinivas:

nat:

geohot:

lilianweng:

Created 5 years ago

Updated 1 year ago

webdataset by webdataset

High-performance I/O system for large deep learning problems, strong PyTorch support

soldni:

chiphuyen:

hammer:

theophilegervet:

Created 6 years ago

Updated 3 months ago

hummingbird by microsoft

Compiler for trained ML models into tensor computation

tgaddair:

chiphuyen:

parano:

ankane:

Created 5 years ago

Updated 2 months ago

RAdam by LiyuanLucasLiu

Optimizer for neural network training, addressing adaptive learning rate variance

danielhanchen:

millionintegrals:

jwyang:

hiyouga:

Created 6 years ago

Updated 4 years ago

transformers by huggingface

ML library for pretrained model inference and training

clmnt:

lilianweng:

karpathy:

tjbck:

Created 6 years ago

Updated 12 hours ago

pytorch-cpp-rl by Omegastick

RL framework using the PyTorch C++ frontend

apaszke:

soumith:

thomwolf:

Created 6 years ago

Updated 5 years ago

neural-mmo by openai

Massively multiagent game environment for training and evaluating intelligent agents

evhub:

jiamings:

millionintegrals:

apsdehal:

Created 6 years ago

Updated 2 years ago

gpt-2 by openai

Code for research paper "Language Models are Unsupervised Multitask Learners"

aravindsrinivas:

simonw:

0hq:

shyamal-anadkat:

Created 6 years ago

Updated 1 year ago

rl-baselines-zoo by araffin

RL agent collection using Stable Baselines (note: unmaintained, use RL-Baselines3 Zoo)

Created 6 years ago

Updated 2 years ago

vel by MillionIntegrals

Deep-learning research library for modular model construction

millionintegrals:

soumith:

nottombrown:

jachiam:

Created 7 years ago

Updated 2 years ago

robotics-rl-srl by araffin

RL/SRL research toolbox for robotics, evaluating state representation learning via RL

dennybritz:

Created 7 years ago

Updated 4 years ago

pytorch-a2c-ppo-acktr-gail by ikostrikov

PyTorch implementations of reinforcement learning algorithms

theophilegervet:

jachiam:

millionintegrals:

glample:

Created 8 years ago

Updated 3 years ago

mujoco-py by openai

Python bindings for MuJoCo physics engine

lilianweng:

aravindsrinivas:

ogabrielluiz:

evhub:

Created 9 years ago

Updated 1 year ago

pytorch-tutorial by yunjey

PyTorch tutorial for deep learning researchers

tjbck:

bryanhelmig:

stas00:

haotian-liu:

Created 8 years ago

Updated 2 years ago

examples by pytorch

PyTorch examples for diverse AI tasks

lilianweng:

hammer:

tjbck:

ari-holtzman:

Created 9 years ago

Updated 2 weeks ago

Feedback? Help us improve.