GitHub · 灵魂体仓库·· 8 天前AI 评分43
AstrAI:轻量级端到端 Transformer 训练与推理框架开源上架 GitHub
ViperEkura/AstrAI
来源与原文
GitHub · 灵魂体仓库
发布时间:(北京时间)
阅读原始来源 · github.com ↗完整链接与内容说明
https://github.com/ViperEkura/AstrAI
本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。
AI 导读
AstrAI 在 GitHub 上开源,是一个覆盖模型构建、训练、评估到服务全生命周期的轻量级 Transformer 框架,基于 PyTorch,要求 Python 3.12+ 并固定 PyTorch 2.13.0。它支持 GQA、MLA、MoE、RoPE 等模型组件,提供预训练、SFT、DPO 与在线 RL(GRPO/PPO)训练,以及 continuous batching、paged KV cache 推理和兼容 OpenAI/Anthropic 协议的 FastAPI 服务。评测工具涵盖 MMLU、HumanEval、IFEval、ROUGE 等,并附 1B checkpoint 下载与 Docker 部署方案。
来源:GitHub · 灵魂体仓库 · github.com