batch-eigen-decomposition topic
List
batch-eigen-decomposition repositories
torch-vectorized
20
Stars
3
Forks
Watchers
Fast analytical implementation of batch eigen-decomposition for 3x3 symmetric matrices with Pytorch. > 250x faster than regular Pytorch implementation of batch eigen-decomposition on GPU.