Does plaidml support only keras or does it support tensorflow also?

Question

I have AMD GPU and wanted to train some neural networks. I have installed Plaidml and it worked with Keras. But for another application I have to use Tensorflow 2.0 for trai