CUDA Python: Performance meets Productivity

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

C++ and Python support for the CUDA Quantum programming model for heterogeneous quantum-classical workflows

[CVPR 2024] Deformable Convolution v4

[CVPR 2023 Highlight] InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutions

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

PyTorch reimplementation of the paper "Swin Transformer V2: Scaling Up Capacity and Resolution" (CVPR 2022)

Code release for DynamicTanh (DyT)

TorchCFM: a Conditional Flow Matching library

PyTorch implementation of FractalGen https://arxiv.org/abs/2502.17437

GPU-accelerated Monte Carlo simulations of 2D Ising Model

Drop in a screenshot and convert it to clean code (HTML/Tailwind/React/Vue)

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

🤗 smolagents: a barebones library for agents that think in python code.

Bayesian Modeling and Probabilistic Programming in Python

PyTorch implementation of Glow

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

无需情感标注的情感可控语音合成模型，基于VITS

Pedagogical example of a simple lattice field theory simulation.

[ICLR 2025] Pyramidal Flow Matching for Efficient Video Generative Modeling

Official Implementation of Rectified Flow (ICLR2023 Spotlight)

从零手搓Flow Matching（Rectified Flow）

code for "Riemannian Flow Matching on General Geometries".

This repository is the official code source of PPFlow: Target-Aware Peptide Design with Torsional Flow Matching (ICML2024)

The aim of this repository is to test and implement Flow-Matching-based models

A minimal PyTorch implementation of Flow Matching for Generative Modeling.

My take on Flow Matching