tropical-gemm

tropical-gemm es una biblioteca de Rust diseñada para la multiplicación de matrices tropicales de alto rendimiento. Es particularmente útil para algoritmos de teoría de gráficos, como problemas de camino más corto y más largo, y para redes neuronales tropicales. Las capacidades clave incluyen: - Múltiples Semirings: Soporte para las estructuras algebraicas MaxPlus, MinPlus y MaxMul. - Aceleración de Hardware: Implementaciones de CPU optimizadas usando SIMD (AVX-512, AVX2, SSE4.1, NEON) y un backend de CUDA acelerado por GPU. - Seguimiento de Argmax: Soporte integrado para el seguimiento de valores argmax, esencial para la retropropagación en redes neuronales tropicales. - Integración con Python: Proporciona enlaces para NumPy y PyTorch, permitiendo una integración perfecta en flujos de trabajo de aprendizaje automático basados en Python con el cálculo automático de gradientes.