The model code is comparatively tiny compared to pytorch or CUDA itself. Translating models from CUDA/C could be laborious but not a barrier.
Making AMD work effortlessly with pytorch et al should make the switch transparent.
Making AMD work effortlessly with pytorch et al should make the switch transparent.