返回全部动态
MLX v0.31.1 发布:CUDA 优化与混合分片支持
原标题:v0.31.1
AI 摘要
MLX 发布了 v0.31.1 补丁版本,包含多项 CUDA 内核优化(如量化 GEMV、Hadamard 变换、SegmentedMM)、混合分片(Hybrid sharding)支持、bug 修复及文档改进。该版本提升了 GPU 上的计算性能与稳定性,并新增了 bartlett 函数。
以上摘要由 AI 生成,可能存在误差。事实请以原文为准。
正文节选
## What's Changed * Bump the patch version by @angeloskath in https://github.com/ml-explore/mlx/pull/3185 * Skip Hopper-only kernels in CI by @zcbenz in https://github.com/ml-explore/mlx/pull/3184 * [CUDA] Fsdp (easy) by @nastya236 in https://github.com/ml-explore/mlx/pull/3130 * Fix ref leak in mx.save/load with file like object by @aisk in https://github.com/ml-explore/mlx/pull/3187 * Fix/missing libs in docs by @ChristophePRAT in https://github.com/ml-explore/mlx/pull/3190 * feat: addin
发布时间:2026-03-12 14:58
抓取时间:2026-08-02 00:31
来源机构:Apple