torch-directml (pypi): A DirectML backend for hardware acceleration in PyTorch.
Keywords: pytorch|plugin|machine learning|directml|directx|d3d12