共 6 个 skill · 数据来自 GitHub 与精选合集,每日更新
NousResearch/hermes-agent/optional-skills/mlops/flash-attention
加速长序列 Transformer 的训练与推理。
NousResearch/hermes-agent/optional-skills/mlops/tensorrt-llm
High-throughput LLM inference on NVIDIA GPUs.EN
Orchestra-Research/AI-Research-SKILLs/12-inference-serving/tensorrt-llm
用 NVIDIA TensorRT 优化 LLM 推理达最大吞吐与最低延迟。在 NVIDIA GPU(A100/H100)生产部署、需比 PyTorch 快 10-100× 的推理、或用量化(FP8/INT4)、在途批处理、多 GPU 扩展服务模型时使用。
Orchestra-Research/AI-Research-SKILLs/06-post-training/miles
用 miles(slime 的生产就绪 fork)做企业级 RL 训练指引。用 FP8/INT4 训练大 MoE 模型、需训练-推理对齐、或需推测性 RL 达最大吞吐时使用。
Orchestra-Research/AI-Research-SKILLs/10-optimization/flash-attention
用 Flash Attention 优化 transformer 注意力,2-4× 加速、10-20× 省内存。训练/跑长序列(>512 token)transformer、注意力遇 GPU 内存问题、或需更快推理时使用。支持 PyTorch 原生 SDPA、flash-attn 库、H100 FP8、滑动窗口注意力。
Orchestra-Research/AI-Research-SKILLs/08-distributed-training/deepspeed
DeepSpeed 分布式训练专家指引——ZeRO 优化阶段、流水线并行、FP16/BF16/FP8、1-bit Adam、稀疏注意力。