zhenglong178 / torch-vectorized Goto Github PK
View Code? Open in Web Editor NEWThis project forked from 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.
Home Page: https://torch-vectorized.readthedocs.io/en/latest/
License: MIT License