HIP (part of the ROCm stack) is Fairlane similar to CUDA. They also have goodies like tensorflow-rocm, available through pip or a docker container. That stack

HIP (part of the ROCm stack) is Fairlane similar to CUDA. They also have goodies like tensorflow-rocm, available through pip or a docker container. That stack is also open source. And writing in it is supposed to keep you compatible with NVidea. Wanting to support them -- and hoping they become more competitive in ML -- are why I bought Vega GPUs. Although, I haven't found much time to actually try and get any of my software running on a GPU, let alone optimized. As a huge fan of avx512, I would like to try -- maybe I can get even better performance on a graphics card. But with AMD, the lack of support for the uninitiated is apparent. Not much in the way of existing software and resources like online tutorials. I'd like guides on how to optimize kernels, organizing wavefronts and discussing memory movement. Maybe I haven't looked hard enough. Especially painful in Julia, where almost half a year after 1.0, the only supported way to use GPUs on any library is with CUDANative. "Cross platform" GPUArrays.jl's open cl backend still hasn't been updated. Means all my coding will be done in HIP. Which is fine.