MLX Swift v0.32.2 公開 — 分散サポートとCUDAパフォーマンスの向上

新機能と改善点

• MLXへの分散サポートを追加。
• LinuxにおけるCUDA NVCC spawnのパフォーマンスを向上。
• fp8変換ユーティリティを追加。
• 量子化レイヤーへのグローバルスケールサポートを追加。
• Swift APIにcountNonzeroを追加。
• より堅牢な統合テストを組み込み。

バグ修正および動作の変更

• CompiledFunction.lockとevalLock間のデッドロック、およびコンパイラキャッシュにおけるデータレースを修正。
• MLXArray (rawPointer:_:dtype:finalizer:) におけるメモリリークを修正。
• .sameモードにおける、偶数カーネルに対するconvolveのpadRight動作を修正。
• MLXNN.Poolのパディング幅の問題を修正。
• PythonのMLX実装に合わせるため、tensordotのデフォルト軸およびnanToNumのデフォルト値を更新。
• linspaceを更新してdtype引数を処理できるようにし、整数引数がデフォルトでfloat32になるように変更。
• Python版との一貫性を高めるため、ALiBi、GRU、Adafactor、およびLionを含む、複数のオプティマイザとレイヤーの動作を調整。