ダウンロードとリソース
TensorFlow用のMetalプラグイン
TensorFlow用のMetalプラグイン1.0がリリースされました。このプラグインは、簡単にインストールすることができ、グラフ最適化機能を備えています。また、16ビット浮動小数点データタイプと、新たに導入された16ビットBrain浮動小数点データタイプを使用した混合精度もサポートしています。
Mac上でのPyTorchトレーニング
演算子とネットワークの幅広いサポート、テスト機能の改善、パフォーマンスの最適化を実現した、MPSバックエンドのベータ版がリリースされました。プロファイリング、カスタムカーネル、混合精度などの新機能により、より多くのMLモデルを高速化して最適化することが可能です。
高速化したMac上でのJAXトレーニング
JAX用の新しいMetalプラグインは、OpenXLAコンパイラとPjRTランタイムを使用して、MacプラットフォームでのJAXによる機械学習のワークロードをGPUで高速化します。