most people I know who have worked at Amazon hated it because of how limited of a scope each team has and the extremely slow pace to get anything approved or deployed. You may have different luck depending on what team you’re on.
I recommend you join a start up, specifically in reinforcement learning as a service, human data, or a neo lab. This will give you direct visibility into the current frontier tech and allow you to learn and grow the most because there will always be things to do.