Hacker News Reader

Tensorflow can work with ROCM which is equivalent to Nvidia's CUDA.

There is no official support, only AMD's custom TF branch. Further ROCM itself works only on Linux. I believe most people writing Swift are using MacOS.

And Julia can do custom ROCM codegen, either by itself or through array and kernel abstractions :) https://github.com/JuliaGPU/AMDGPU.jl

https://juliagpu.gitlab.io/KernelAbstractions.jl/

https://github.com/JuliaGPU/GPUArrays.jl/pulse

Indeed, the Julia scene looks enticing right now. If only AMD ROCM supported MacOS... :/

Reply on news.ycombinator.com