torchvtk

torchvtk is a PyTorch-based framework for efficient loading, caching and transformation of volumetric data, addressing common challenges like IO bottlenecks. I started this project with a colleague and used it for my research. The project is no longer in development as many alternatives emerged in the recent years.