Tensors and Dynamic neural networks in Python with strong GPU acceleration
Modified version of PyTorch able to work with changes to GPGPU-Sim
PyTorch version of MeshSegNet for tooth segmentation of intraoral scans (point cloud/mesh). The code also includes visdom for training visualization.