Nvidia Warp - 高性能GPUシミュレーションとグラフィックスのためのPythonフレームワーク
(github.com/NVIDIA)- 高性能なシミュレーションおよびグラフィックスコードを作成するためのPythonフレームワーク
- Python関数をJITコンパイルし、CPUまたはGPUで実行可能な効率的なカーネルコードへ変換
- 空間コンピューティング向けに設計されており、物理シミュレーション、ロボティクス、幾何処理などのための多様なプリミティブを提供
- Warpカーネルは微分可能で、PyTorchやJAXのような機械学習パイプラインの一部として利用可能
- 要件
- Python 3.9以上を推奨
- Windows、Linux、macOSでx86-64およびARMv8 CPUをサポート
- GPUサポートにはCUDA対応のNVIDIA GPUおよびドライバーが必要（最低 GeForce GTX 9xx）
- PyPIからインストール可能
- CUDA 12.5ランタイムでビルドされたバイナリはGitHub Releasesページで提供
まだコメントはありません。