xinwangChen
Follow
Stars
PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.
PyTorch implementation of Data Free Quantization Through Weight Equalization and Bias Correction.
[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion
End-to-End Object Detection with Transformers
Protocol Buffers - Google's data interchange format
Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
Neighborhood Attention Extension. Bringing attention to a neighborhood near you!
VMamba: Visual State Space Models，code is based on mamba
[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
The suite of modeling video with Mamba
A diffusion-based framework for spatio-temporal point processes
Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
OpenDiT: An Easy, Fast and Memory-Efficient System for DiT Training and Inference
OTOv1-v3, NeurIPS, ICLR, TMLR, DNN Training, Compression, Structured Pruning, Erasing Operators, CNN, Diffusion, LLM
MNN is a blazing fast, lightweight deep learning framework, battle-tested by business-critical use cases in Alibaba
Two time-scale update rule for training GANs
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".
[ECCVW 2022] The codes for the work "Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentation"
Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
PyTorch code and models for V-JEPA self-supervised learning from video.
Official codebase for I-JEPA, the Image-based Joint-Embedding Predictive Architecture. First outlined in the CVPR paper, "Self-supervised learning from images with a joint-embedding predictive arch…
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"