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.

github.com/banctilrobitaille/torch-vectorized

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.