If they get TensorFlow/Torch/MXNet running with no changes in the code, 32GB HBM2 for training models would be awesome! These days hitting memory limitations is biting more than general slowness of GPUs during Deep Learning training.

Hm? not close enough?

https://github.com/ROCmSoftwarePlatform/tensorflow

https://github.com/ROCmSoftwarePlatform/cutorch_hip

https://github.com/ROCmSoftwarePlatform/mxnet

https://github.com/ROCmSoftwarePlatform/hipCaffe

Is TF running on AMD Vega's? Keras?

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