juliusbk·geçen yıl·discussYeah, I remember looking into it myself as well, and not finding any easy path. A shame.... Maybe there's a hard way to do it though :)
juliusbk·geçen yıl·discussThis looks super cool! Looking forward to trying it.I think a killer feature of these gpu-plotting libraries would be if they could take torch/jax cuda arrays directly and not require a (slow) transfer over cpu.