147 citations · 741 across the 35 of their papers we have counts for
5 papers · 1 filter
CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforcement Learning
Songqiao Su, Xiaoya Li, Albert Wang +3
In this paper, we propose CUDA-L2, a system that combines large language models (LLMs) and reinforcement learning (RL) to automatically optimize Half-precision General Matrix Multi…
CRINN: Contrastive Reinforcement Learning for Approximate Nearest Neighbor Search
Xiaoya Li, Albert Wang, Guoyin Wang +2
Approximate nearest-neighbor search (ANNS) algorithms have become increasingly critical for recent AI applications, particularly in retrieval-augmented generation (RAG) and agent-b…
CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learning
Xiaoya Li, Albert Wang, Guoyin Wang +2
The exponential growth in demand for GPU computing resources has created an urgent need for automated CUDA optimization strategies. While recent advances in LLMs show promise for c…
FaceID-6M: A Large-Scale, Open-Source FaceID Customization Dataset
Shuhe Wang, Xiaoya Li, Jiwei Li +8
Due to the data-driven nature of current face identity (FaceID) customization methods, all state-of-the-art models rely on large-scale datasets containing millions of high-quality…
Turn That Frown Upside Down: FaceID Customization via Cross-Training Data
Shuhe Wang, Xiaoya Li, Xiaofei Sun +4
Existing face identity (FaceID) customization methods perform well but are limited to generating identical faces as the input, while in real-world applications, users often desire…