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

Call python from fortran

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 -> …