I may be missing something, but the only machine learning focused api I know of are AMD’s ROCM, Nvidia’s CUDA, and now Intel’s oneAPI. I haven’t looked into Apple’s machine learning frameworks and I consider vulkan more of a general purpose api than a machine learning one.