pytorch

強力なGPUアクセラレーションを備えたPythonでのテンソルと動的ニューラルネットワーク。

Vision、Text、Reinforcement Learningなどにおけるpytorchのサンプル集

コンピュータビジョンに特化したデータセット、トランスフォーム、およびモデル

PyTorch のチュートリアル。

生成AIモデルの訓練のためのPyTorchネイティブプラットフォーム

PyTorchのためのモバイル、組み込み、エッジを横断するオンデバイスAI。

PyTorchにおけるニューラルネットワークのトレーニングと評価を柔軟かつ透過的に支援する高レベルライブラリ。

強化学習のための、モジュール式でプリミティブ中心の Python ファーストな PyTorch ライブラリ

TensorRTを使用したNVIDIA GPU向けPyTorch/TorchScript/FXコンパイラ

トレーニングおよび推論のための PyTorch ネイティブ量子化

PyTorchを搭載した、音声信号処理のためのデータ操作および変換ツール

XLA デバイス（Google TPU など）での PyTorch の有効化

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

マルチマシーン学習用のさまざまなプリミティブを備えた集合通信ライブラリ

https://pytorch.org/hub/ への提出物

PyTorchのC++拡張

CPU情報取得ライブラリ（x86/x86-64/ARM/ARM64、Linux/Windows/Android/macOS/iOS対応）

PyTorch のパフォーマンスを評価するために使用されるオープンソースベンチマークのコレクション。

TensorDict は PyTorch 専用のテンソルコンテナです。

タイムラインのトレースとハードウェアパフォーマンスカウンターへのアクセスを提供する CPU+GPU プロファイリングライブラリ

最小限のボイラープレートで高速かつスケーラブルなMLカーネルを簡単に記述できるPython組み込みDSL