Train neural networks using AMD GPUs and Keras