Wei, Jia

9 publications

CVPRW 2025 Mixture-of-Shape-Experts (MoSE): End-to-End Shape Dictionary Framework to Prompt SAM for Generalizable Medical Segmentation Jia Wei, Xiaoqi Zhao, Jonghye Woo, Jinsong Ouyang, Georges El Fakhri, Qingyu Chen, Xiaofeng Liu

ICLRW 2025 SageAttention2: Efficient Attention with Smoothing Q and Per-Thread Quantization Jintao Zhang, Haofeng Huang, Pengle Zhang, Jia Wei, Jun Zhu, Jianfei Chen

ICML 2025 SageAttention2: Efficient Attention with Thorough Outlier Smoothing and Per-Thread INT4 Quantization Jintao Zhang, Haofeng Huang, Pengle Zhang, Jia Wei, Jun Zhu, Jianfei Chen

NeurIPS 2025 SageAttention3: Microscaling FP4 Attention for Inference and an Exploration of 8-Bit Training Jintao Zhang, Jia Wei, Haoxu Wang, Pengle Zhang, Xiaoming Xu, Haofeng Huang, Kai Jiang, Jianfei Chen, Jun Zhu

ICLR 2025 SageAttention: Accurate 8-Bit Attention for Plug-and-Play Inference Acceleration Jintao Zhang, Jia Wei, Pengle Zhang, Jun Zhu, Jianfei Chen

ICML 2025 SpargeAttention: Accurate and Training-Free Sparse Attention Accelerating Any Model Inference Jintao Zhang, Chendong Xiang, Haofeng Huang, Jia Wei, Haocheng Xi, Jun Zhu, Jianfei Chen

ICLRW 2025 SpargeAttn: Training-Free Sparse Attention Accelerating Any Model Inference Jintao Zhang, Chendong Xiang, Haofeng Huang, Jia Wei, Haocheng Xi, Jun Zhu, Jianfei Chen

IJCAI 2022 Unsupervised Multi-Modal Medical Image Registration via Discriminator-Free Image-to-Image Translation Zekang Chen, Jia Wei, Rui Li

CVPRW 2021 Pixel-Guided Dual-Branch Attention Network for Joint Image Deblurring and Super-Resolution Si Xi, Jia Wei, Weidong Zhang