torchvtk

torchvtk is a PyTorch-based framework for efficient volume data loading, queued caching of data, native PyTorch-based transformations and general utilities to work with volume data and transfer functions. I’m the main developer of the framework.