Depending on how deep you want to go in the implementation details, you may want to look into research positions. An MLE job isn't that different from a backend

Depending on how deep you want to go in the implementation details, you may want to look into research positions. An MLE job isn't that different from a backend dev job.