Not with mainline kernel (which supports up to Polaris), but it it already got merged into 4.18 that is going to be released in a couple of weeks. Until then it requires a bit extra effort https://github.com/RadeonOpenCompute/ROCm#dkms-driver-instal...

> Keras?

over opencl with https://github.com/plaidml/plaidml