跳到正文
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