Amazon's comp doesn't distinguish AI/ML comp vs. general software engineering officially, but it's possible that those teams have a higher budget, given the competitiveness of the skillset. Levels posted a blog post with details on how much AI engineers are getting paid compared to their colleagues and, while it's definitely higher, it's still not quite 2x or 1.5x higher. https://www.levels.fyi/blog/ai-engineer-compensation.html