GitHub repository: sarvex/tensorflow-directml-plugin
Description: DirectML PluggableDevice plugin for TensorFlow 2