pytorch-directml (pypi): Tensors and Dynamic neural networks in Python with strong GPU acceleration
Keywords: pytorch|machine|learning|directml|directx|d3d12