À propos du projet

tropical-gemm est une bibliothèque Rust conçue pour une multiplication matricielle tropicale (ou max-plus/min-plus) à haute performance. Elle est particulièrement utile pour les algorithmes de la théorie des graphes, tels que les problèmes de chemin le plus court et le plus long, ainsi que pour les réseaux de neurones tropicaux. Les capacités clés comprennent : - Plusieurs semianneaux : Support des structures algébriques MaxPlus, MinPlus et MaxMul. - Accélération matérielle : Implémentations CPU optimisées utilisant SIMD (AVX-512, AVX2, SSE4.1, NEON) et un backend CUDA accéléré par GPU. - Suivi Argmax : Support intégré pour le suivi des valeurs argmax, essentiel pour la rétropropagation dans les réseaux de neurones tropicaux. - Intégration Python : Fournit des liaisons pour NumPy et PyTorch, permettant une intégration transparente dans les flux de travail d'apprentissage automatique basés sur Python avec calcul de gradient automatique. La bibliothèque est écrite en Rust pour la performance et la sécurité, et elle offre une API propre pour les développeurs Rust et Python.