返回全部动态

MLX v0.32.4 发布:新增 VJP 支持与多项修复

原标题:v0.32.4

MLX Releases一手来源开源质量 74

AI 摘要

MLX 发布 v0.32.4 版本,这是一个以修复和优化为主的补丁更新。主要变更包括为缩放点积注意力(SDPA)和门控 delta 网络添加 VJP 支持、新增 Python 3.15 支持、修复大量 Metal/CUDA 后端与视图/步幅相关的 bug,并优化了量化矩阵乘法和 gather_qmm 等内核性能。该版本提升了 MLX 在 Apple Silicon 及 CUDA 上的数值稳定性与推理效率。

以上摘要由 AI 生成,可能存在误差。事实请以原文为准。

正文节选

## What's Changed * Adding VJP for the scaled dot product attention. by @tpegolotti in https://github.com/ml-explore/mlx/pull/4563 * python: Add __pos__ support to array by @aaishwarymishra in https://github.com/ml-explore/mlx/pull/4488 * Add Python 3.15 and 3.15t support by @XXXXRT666 in https://github.com/ml-explore/mlx/pull/4555 * Add fixed size `unique` operation by @louen in https://github.com/ml-explore/mlx/pull/4501 * Fix normalizing single-element slice with negative stride by @axi


发布时间:2026-10-15 07:37
抓取时间:2026-10-15 10:14
来源机构:Apple
阅读原文github.com