Use PyTorch instead of NumPy #10

Merged
cfalas merged 5 commits from torch into main 2025-01-30 13:58:26 +02:00

5 Commits

Author SHA1 Message Date
Christos Falas
638440feff fix heatmap generation 2025-01-30 13:58:08 +02:00
Christos Falas
785021b557 Use eigh instead of eig
The mean of Hermitian matrices is also Hermitian
2025-01-30 13:58:08 +02:00
Christos Falas
42ac105240 Improve logging 2025-01-30 13:58:08 +02:00
Christos Falas
c2963f4a9f Make heatmap on GPU
Still has some weird artifacts, some axis ordering might be incorrect
2025-01-30 13:58:08 +02:00
Christos Falas
a1e5f2fec7 set up torch AoA estimation 2025-01-30 13:58:08 +02:00