The initial preview of TensorFlow-DirectML-Plugin has been released as a PluggableDevice for TensorFlow 2.
The DirectML backend is available to accelerate training on any DX12 compatible GPU.
Pick up the package on PyPI today and leave any feedback on GitHub.
Learn more ↓