Stars
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Python Implementation of Reinforcement Learning: An Introduction
[CVPR 2024] MagicAnimate: Temporally Consistent Human Image Animation using Diffusion Model
Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization.
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"
A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.
PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
SuperGlue: Learning Feature Matching with Graph Neural Networks (CVPR 2020, Oral)
Vector (and Scalar) Quantization, in Pytorch
[SIGGRAPH'24] 2D Gaussian Splatting for Geometrically Accurate Radiance Fields
pytorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation" https://arxiv.org/abs/1612.00593
[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model
Learning Continuous Signed Distance Functions for Shape Representation
State-of-the-art, simple, fast unbounded / large-scale NeRFs.
[CVPR'24 Best Student Paper] Mip-Splatting: Alias-free 3D Gaussian Splatting
Implementation of ConvMixer for "Patches Are All You Need? 🤷"
Calculate signed distance fields for arbitrary meshes
Tooling for the Common Objects In 3D dataset.
improves over nerf in 360 capture of unbounded scenes
BARF: Bundle-Adjusting Neural Radiance Fields 🤮 (ICCV 2021 oral)