Tensors and Dynamic neural networks in Python with strong GPU acceleration
On-device AI across mobile, embedded and edge for PyTorch