Thanks! The current version requires around 18GB of total GPU memory. Right now I am heavily bottlenecked by the memory transfer costs, but there is some low-ha

Thanks! The current version requires around 18GB of total GPU memory. Right now I am heavily bottlenecked by the memory transfer costs, but there is some low-hanging fruit to amortize them by doing much more compute per snapshot transfer. After that, I'm very curious to see what the profiler will show.