← Search

Haohui Mai

3 accepted papers

2026

FSA: An Alternative Efficient Implementation of Native Sparse Attention Kernel

ICLR 2026poster

Recent advance in sparse attention mechanisms has demonstrated strong potential for reducing the computational cost of long-context training and inference in large language models (LLMs). Native Sparse Attention (NSA), one state-of-the-art approach, introduces natively trainable, hardware-aligned sp…

Cited by 0SourcecodeScholar
2025

Training Language Models to Generate Quality Code with Program Analysis Feedback

NeurIPS 2025poster

Code generation with large language models (LLMs), often termed vibe coding, is increasingly adopted in production but fails to ensure code quality, particularly in security (e.g., SQL injection vulnerabilities) and maintainability (e.g., missing type annotations). Existing methods, such as supervis…

Cited by 0SourcecodeScholar