torch

2.8.0
1.11G

Tensors and Dynamic neural networks in Python with strong GPU acceleration