Applied Scientist • Mid-Level

Who you are

You have 3+ years of experience building machine learning models, demonstrating a strong foundation in AI and applied science. Your proficiency in programming languages such as Python, Java, or C++ allows you to tackle complex problems effectively. You possess a solid understanding of deep learning methods and tools, particularly with frameworks like PyTorch and JAX, which you have used in previous projects. Your academic background includes a PhD or Master's degree in computer science or a related field, equipping you with the theoretical knowledge to support your practical skills. You have a strong background in scientific research, showcasing your ability to generate and implement innovative ideas in machine learning. You are willing to step outside typical role boundaries to achieve results, demonstrating a proactive approach to problem-solving.

What you'll do

In this role, you will contribute directly to the development of AI agents in an applied research capacity. Your responsibilities will include model training, dataset design, and pre- and post-training optimization, ensuring that the models perform effectively in both virtual and physical environments. You will collaborate with a small, talented team to push the boundaries of what AI can achieve, particularly in the context of Amazon's AGI organization. Your work will involve combining large language models with reinforcement learning to solve complex reasoning and planning tasks. You will have the autonomy to explore high-risk, high-payoff research avenues, contributing to the foundational capabilities of useful AI agents. Your insights and contributions will play a crucial role in shaping the future of AI at Amazon.

What we offer

Amazon provides a dynamic work environment where innovation is encouraged and supported. You will have access to a range of resources and the opportunity to work on cutting-edge technology that has the potential to redefine AI capabilities. The company offers a comprehensive benefits package, including equity options, sign-on payments, and a full range of medical and financial benefits. You will be part of a culture that values collaboration and creativity, allowing you to grow both personally and professionally as you contribute to impactful projects.