banctilrobitaille / torch-vectorized

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.
21Updated 3 years ago

Alternatives and similar repositories for torch-vectorized:

Users that are interested in torch-vectorized are comparing it to the libraries listed below