PyTorch is an optimized tensor library for deep learning using GPUs and CPUs. https://www.learnpytorch.io/