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