Starred repositories
Generative Modeling via Drifting in MLX
Hybrid ML + physics model of the Earth's atmosphere
Python package for easily computing climate feedbacks using radiative kernels.
PyTorch code and models for VJEPA2 self-supervised learning from video.
Official PyTorch implementation of the paper "RWKV-UNet: Improving UNet with Long-Range Cooperation for Effective Medical Image Segmentation"
Low cost motion capture system for room scale tracking
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), ga…
A unified framework for Deep Learning Models on tabular data
TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.
[TMLR 2025] Thera: Aliasing-Free Arbitrary-Scale Super-Resolution with Neural Heat Fields
An optimal transport-based algorithm for inferring cell trajectories of spatial transcriptomics.
[CVPR 2024 Highlight] Official PyTorch implementation of SpatialTracker: Tracking Any 2D Pixels in 3D Space
Official pytorch repository for "Diffusion Posterior Sampling for General Noisy Inverse Problems"
An official pytorch implementation of "Parallel Diffusion Model of Operator and Image for Blind Inverse Problems" (CVPR 2023)
Text to Image Latent Diffusion using a Transformer core
Extract Satellite Imagery from public constellations at scale
End-to-End Object Detection with Transformers
Fence GAN: Towards Better Anomaly Detection
PyTorch Implementations for DeeplabV3 and PSPNet
DeepFill v1/v2 with Contextual Attention and Gated Convolution, CVPR 2018, and ICCV 2019 Oral
PyTorch implementation of Convolutional Recurrent Neural Network
A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.
Detection and Tracking of Tropical Cyclones
Collection of generative models in Pytorch version.
Latex code for making neural networks diagrams
Unofficial PyTorch implementation of the paper titled "Progressive growing of GANs for improved Quality, Stability, and Variation"
Progressive Growing of GANs for Improved Quality, Stability, and Variation
🔥🔥 PyTorch implementation of "Progressive growing of GANs (PGGAN)" 🔥🔥
Implementation (with some experimentation) of the paper titled "VARIATIONAL DISCRIMINATOR BOTTLENECK: IMPROVING IMITATION LEARNING, INVERSE RL, AND GANS BY CONSTRAINING INFORMATION FLOW" (arxiv -> …