Tensors and Dynamic neural networks in Python with strong GPU acceleration
BLAS/LAPACK backend (openblas on linux/windows, Accelerate on macOS). On by default; disable with [core,...] when using [mkl] instead.
Build with CUDA GPU backend
Use distributed training/inference (Gloo, MPI, libuv, TensorPipe)
Build with gflags
Build with glog
Build with LLVM
Intel performance backend: use oneMKL for BLAS/LAPACK and enable oneDNN (MKLDNN) CPU acceleration via ideep. Mutually exclusive with [blas] (the openblas default-feature) — install as libtorch[core,mkl,...] to skip the openblas backend, otherwise both backends get built and only MKL is linked.
Build with Vulkan GPU backend
v2.12.0#0
(windows & !static) | osx | linux
Complex license
Manifest