项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化，观测越久曲线越完整
数据来源：GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 149
技术标签
核心亮点
- 提供简洁的 API，几行代码即可实现数据并行，降低上手门槛
- 内置梯度累积和混合精度，方便训练大模型时节省显存
- 支持自定义分片策略，灵活适配不同并行模式
不足之处
- 早期项目，文档和示例相对较少，社区生态待完善
- 功能覆盖有限，高级并行特性可能不如成熟框架全面
适用场景
- 在 JAX 中快速实现多 GPU 数据并行训练
- 训练参数量较大的模型时使用梯度累积和混合精度
- 研究原型中需要灵活配置分片策略的分布式实验
替代项目
JAX's pmap、Flax、Alpa
上一篇：JetStream
下一篇：FlexPrefill
同类项目推荐
MyScaleDB
开源
用SQL同时搞定向量搜索和全文检索，AI应用数据底座一步到位
A @ClickHouse fork that supports high-performance vector search and full-text search···
emmy
开源
选配方或自调优，让 LLM 推理在 GPU 上跑得更快
Optimized GPU compiler for LLM inference. Choose from a list of optimized recipes or···
OpenContracts
开源
让智能体读懂文档，一键提取结构化数据。
The open document intelligence platform for builders and hackers - DMS for the agent···
build-and-deploy-real-time-featu···
开源
用Python快速搭建实时特征管道，让模型用上最新数据。
Develop and deploy a real-time feature pipeline in Python, using Bytewax and Hopswo···
ai00_server
开源
一键部署 RWKV 模型，集成聊天、嵌入、RAG 和智能代理。
The all-in-one RWKV runtime box with embed, RAG, AI agents, and more.
NanoLLM
开源
本地跑大模型，量化多模态一条龙
Optimized local inference for LLMs with HuggingFace-like APIs for quantization, visi···
lzy
开源
一套工作流，本地远程无缝跑，算力随便用
Platform for a hybrid execution of ML workflows that transparently integrates local ···
Comfyui-zhenzhen
开源
在 ComfyUI 里拖节点调用平价 AI 生成接口
https://ai.t8star.org/register?aff=dP7j Comfyui的各种平价API调用节点，支持seedance2.···
velobase-harness
开源
把 AI 原型一键变成能收费的 SaaS
All AI developers deserve to get paid. Turn your AI prototype into a paid SaaS with···