JuliaGPU/CUDA.jl

Schur decomposition

Open

#2,077 opened on 2023年9月11日

GitHub で見る
 (1 comment) (0 reactions) (0 assignees)Julia (274 forks)batch import
enhancementhelp wanted

Repository metrics

Stars
 (1,408 stars)
PR merge metrics
 (平均マージ 5d 5h) (30d で 16 merged PRs)

説明

Schur decomposition is implemented efficiently on GPU in PyTorch, Jax, and MatLab. It is an essential ingredient for the efficient computation of matrix square root (sqrtm).

It would be ideal to have this decomposition implemented directly in CUDA.jl. The only alternative I see at the moment is full-on diagonalization which is less efficient and less stable.

コントリビューターガイド