MLXライブラリにCUDAバックエンドを追加するプルリクエスト。CUDAの統一メモリやNVIDIA GPUの普及に着目し、Macでの開発とスーパーコンピューターへのデプロイを容易にすることを目指している。CUDAバックエンドはまだ開発中で、現在はロジスティック回帰の例が動作する段階。Jetsonデバイスでのビルドエラーも報告されている。パフォーマンス改善のため、イベントの実装変更やメモリ使用量削減などの最適化が行われている。現在は個々のオペレーションに分割され、今後の変更はインクリメンタルPRで提出される予定。