Tensors and dynamic neural networks in Python with strong GPU acceleration for deep learning research and production.
PyTorch deep learning framework providing tensor computation with GPU acceleration and automatic differentiation.
Datasets, model architectures, and image transformations for computer vision tasks in PyTorch.