scalax 是基础设施领域的开源项目，由 young-geng 开发，2023 年首次发布。
它收录于基础设施分类，该分类目前共有 1,133 个项目，GitHub 星标数为 149。
项目已超过三个月没有代码更新，维护节奏明显放缓，最近一次代码更新于 2026-08-28。开源库，Apache-2.0 许可证，完全免费，无付费版本。
它主要面向的使用场景是：在JAX中快速实现多GPU数据并行训练。同类可对比的替代方案包括 JAX's pmap、Flax、Alpa。
GitHub 星标
★ 149
维护状态
低维护
是否开源
是
定价模式
free
项目数据
分类开源模型
开发团队young-geng
所属国家
定价模式free
价格说明开源库，Apache-2.0 许可证，完全免费，无付费版本。
访问状态
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型jax,large-language-models
GitHub 星标★ 149
30天Star增速
HF 下载量
上线时间2023-07-25 00:00:00
最近更新2026-08-28 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数2
使用教程
核心亮点
- 提供简洁的 API，几行代码即可实现数据并行，降低上手门槛
- 内置梯度累积和混合精度，方便训练大模型时节省显存
- 支持自定义分片策略，灵活适配不同并行模式
不足之处
- 早期项目，文档和示例相对较少，社区生态待完善
- 功能覆盖有限，高级并行特性可能不如成熟框架全面
适用场景
- 在 JAX 中快速实现多 GPU 数据并行训练
- 训练参数量较大的模型时使用梯度累积和混合精度
- 研究原型中需要灵活配置分片策略的分布式实验
替代项目
JAX's pmap、Flax、Alpa
项目介绍
上一篇：JetStream
下一篇：FlexPrefill
同类项目推荐
freebuff-proxy
开源
聚合多账号，一键接入 OpenAI 兼容 API，轻松管理会话。
[DEPRECATED / UNMAINTAINED] Multi-account gateway. No longer maintained; fork it and···
microduck
开源
用 Rust 造一只会走路的桌面小鸭，快速上手双足机器人。
A Tiny biped duck robot
soperator
开源
用 Kubernetes 原生方式运行 Slurm，简化 HPC 集群管理。
Run Slurm in Kubernetes
ollama
开源
一条命令本地跑起大模型，免费、私密、不卡顿
Get up and running with Kimi, GLM, MiniMax, DeepSeek, gpt-oss, Qwen, Gemma and other···