Improve IPython imports and fix GPU to CPU transfers when using device=cuda - #8

Open

gs512 wants to merge 3 commits into

azizilab:mainfrom

Commits on Oct 9, 2025

adding device to torch.tensor, adding gpu to cpu conversion
gs512
committed
9d5a339
updating fig.canvas.tostring_rgb() to buffer_rgba which includes alpha for transparency
gs512
committed
6e1208a
fixed GPU to CPU for basis during plot_every_k_epochs, and IPython display imports
gs512
committed
357a799