Consider using cupy.linalg.eigh to accelerate SD2 #48
Description
cupy.linalg.eigh may be a drop-in replacement for numpy.linalg.eigh and yield big boost:
https://docs.cupy.dev/en/stable/reference/generated/cupy.linalg.eigh.html
{{ message }}
cupy.linalg.eigh may be a drop-in replacement for numpy.linalg.eigh and yield big boost:
https://docs.cupy.dev/en/stable/reference/generated/cupy.linalg.eigh.html