tropical-gemm

tropical-gemmは、トロピカル（またはmax-plus/min-plus）行列積のための高性能Rustライブラリです。最短経路や最長経路問題などのグラフ理論アルゴリズム、およびトロピカルニューラルネットワークに特に有用です。 主な機能には以下が含まれます： - 複数の半環：MaxPlus、MinPlus、MaxMulの代数構造をサポート。 - ハードウェア加速：SIMD（AVX-512、AVX2、SSE4.1、NEON）を使用した最適化されたCPU実装と、GPU加速のCUDAバックエンド。 - argmaxトラッキング：トロピカルニューラルネットワークのバックプロパゲーションに不可欠なargmax値のトラッキングを内蔵。 - Python統合：NumPyとPyTorchの両方に対するバインディングを提供し、自動勾配計算によるPythonベースの機械学習ワークフローへのシームレスな統合を可能にします。 このライブラリは、パフォーマンスと安全性のためにRustで書かれており、RustとPythonの両方の開発者にクリーンなAPIを提供します。