Not having a layer like this has left AMD completely out of the AI game that has made NVDA the world's most valuable company.
There are also versions of PyTorch, TensorFlow and JAX with AMD support.
PyTorch's torch.compile can generate Triton (OpenAI's GPU compiler) kernels, with Triton also supporting AMD.