TensorFlow introduces the PluggableDevice architecture, which seamlessly integrates accelerators (GPUs, TPUs) with TensorFlow without making any changes in the TensorFlow code. PluggableDevice, as the name suggests, provides plug-in options for registering the devices with TensorFlow. It is constructed using the StreamExecutor C API and builds on the work done for Modular TensorFlow. In TF 2.5, the PluggableDevice feature is available.
Full Story: https://www.marktechpost.com/2021/06/24/tensorflow-introduces-pluggabledevice-architecture-to-integrate-accelerators-gpus-tpus-with-tensorflow-without-making-any-changes-in-the-tensorflow-code/