1 repo
APIs for integrating high-performance C++, CUDA, or SYCL code into the computational graph.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Native Extension Interfaces. Refine with filters or upvote what's useful.
PyTorch is a machine learning framework centered on a GPU-ready tensor library that supports multi-dimensional array operations across both CPU and accelerator hardware. It provides a foundational infrastructure for mathematical computation and dynamic neural network construction, utilizing a tape-based automatic diffe