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