Volumetric structures such as voxels and SDFs implemented in pytorch
Batch multidimensional indexing for pytorch
Generate customized voxel representations of protein-ligand complexes using GPU.