热门

最新

红包

立Flag

投票

同城

我的

发布
devfranoischollet
Dreamweaver183
1 年前
truedevfranoischollet

它还支持模型并行训练和推理,因此您可以轻松使用具有100亿、200亿或1000亿参数的大型模型。

它让您可以非常细致地控制如何在设备之间分片您的权重和中间激活。

CSDN App 扫码分享
分享
评论
33
  • 复制链接
  • 举报
下一条:
在许多情况下,由于采用了XLA编译(现在也默认在TensorFlow的GPU和TPU上开启了),JAX成为了最快的后端。
立即登录