LeetCode for AI Infra:从 RMSNorm、RoPE 到 FlashAttention,在线编写 Python 实现并即时评测。
实现 RMSNorm
实现 Rotary Position Embedding (RoPE)
实现数值稳定的 Softmax
实现 FlashAttention-2 前向
vLLM kv cache 管理