PyTorch

PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.