Levanter

Levanter

开发框架/库 2.3k 免费增值
CRFM 开发的大规模语言模型训练框架,支持高效分布式训练。

搭配玩法

1
Levanter + JAX JAX + XLA+ NVIDIA GB200 NVL72

在GPU集群上训练535B MoE大模型,解决专家并行与通信效率问题

JAX负责数值计算与自动微分,XLA编译到加速器,Levanter为训练框架;因JAX/XLA GPU环境缺现成专家并行方案,团队自行实现EP

✓ 计算与通信可重叠✓ 适合大规模MoE训练

产品简介

CRFM 开发的大规模语言模型训练框架,支持高效分布式训练。

用户评价

还没有评价
这款产品刚收录,成为第一个评价的人
写第一条评价