I feel the need for Eigen-Speed

PyTorch eigenvalues on CUDA can be unexpectedly slow due to device synchronization. This post shows how to call CuPy via DLPack for fast GPU eigvalsh/eigh while keeping gradients for training.