-
Huazhong University of Science and Technology
- Wuhan China
- https://robert-zwr.github.io/
Stars
Efficient Triton Kernels for LLM Training
[CVPR 2024 Highlight] DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Models
Run your own AI cluster at home with everyday devices 📱💻 🖥️⌚
FlagGems is an operator library for large language models implemented in Triton Language.
Public code repo for paper "A Single Transformer for Scalable Vision-Language Modeling"
Implementation of E2-TTS, "Embarrassingly Easy Fully Non-Autoregressive Zero-Shot TTS", in Pytorch
A one-stop, open-source, high-quality data extraction tool, supports PDF/webpage/e-book extraction.一站式开源高质量数据提取工具,支持PDF/网页/多格式电子书提取。
Cambrian-1 is a family of multimodal LLMs with a vision-centric design.
DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model.
Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton
[ICCV 2023] DDColor: Towards Photo-Realistic Image Colorization via Dual Decoders
TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.
A massively parallel, high-level programming language
Hunyuan-DiT : A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understanding
A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
CoreNet: A library for training deep neural networks
[GPT beats diffusion🔥] [scaling laws in visual generation📈] Official impl. of "Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction". An *ultra-simple, user-friendly …
One-step image-to-image with Stable Diffusion turbo: sketch2image, day2night, and more
[ECCV 2024] official code for "Long-CLIP: Unlocking the Long-Text Capability of CLIP"
AniPortrait: Audio-Driven Synthesis of Photorealistic Portrait Animation
Flash Attention in ~100 lines of CUDA (forward pass only)
Official Code for Stable Cascade