Sobre o projeto
tropical-gemm é uma biblioteca Rust projetada para multiplicação de matrizes tropicais (ou max-plus/min-plus) de alto desempenho. Ela é particularmente útil para algoritmos de teoria de gráficos, como problemas de caminho mais curto e mais longo, e para redes neurais tropicais.
Principais recursos incluem:
- Suporte a múltiplos semianéis: Compatibilidade com as estruturas algébricas MaxPlus, MinPlus e MaxMul.
- Aceleração de hardware: Implementações otimizadas para CPU usando SIMD (AVX-512, AVX2, SSE4.1, NEON) e um backend CUDA acelerado por GPU.
- Rastreamento de argmax: Suporte integrado para rastrear valores de argmax, essencial para backpropagation em redes neurais tropicais.
- Integração com Python: Fornece bindings para NumPy e PyTorch, permitindo integração perfeita em workflows de machine learning baseados em Python, com cálculo automático de gradientes.
Comments
0 Rating appears after 10 ratings
Sign in to join the discussion.