PyTorch: Tensors and Dynamic neural networks in Python with strong GPU acceleration.
2.4.1
aarch64-darwin
aarch64-linux
x86_64-linux