Machine Learning Engineer • Senior

Who you are

You have 5+ years of experience in software engineering with a strong focus on machine learning and AI applications — you've successfully developed and deployed machine learning models that impact real-world applications. Your expertise in AWS services, particularly AWS Neuron, allows you to leverage cloud capabilities to enhance model performance and scalability.

You possess deep knowledge of Python and popular machine learning frameworks such as TensorFlow and PyTorch — your coding skills enable you to write efficient, maintainable code that adheres to best practices. You are comfortable working with large datasets and have experience in performance tuning and optimization of machine learning models.

You thrive in collaborative environments and enjoy working closely with cross-functional teams — your strong communication skills help you articulate complex technical concepts to non-technical stakeholders. You are passionate about innovation and continuously seek to improve processes and technologies.

Desirable

Experience with large language models like GPT-2 and GPT-3 is a plus — you understand the intricacies of training and deploying these models at scale. Familiarity with distributed training techniques and tools will set you apart as a candidate.

What you'll do

In this role, you will be responsible for designing and implementing machine learning solutions that leverage AWS Neuron and other AWS services — your work will directly contribute to the development of scalable applications that serve millions of users. You will collaborate with product managers and other engineers to define requirements and translate them into technical specifications.

You will conduct experiments to evaluate model performance and iterate on designs based on data-driven insights — your analytical mindset will help you identify areas for improvement and implement solutions that enhance model accuracy and efficiency. You will also mentor junior engineers, sharing your knowledge and expertise to foster a culture of learning within the team.

You will participate in code reviews and contribute to architectural discussions, ensuring that the systems you build are robust and maintainable — your attention to detail will help prevent issues before they arise. Additionally, you will stay up-to-date with the latest advancements in machine learning and AI, applying new techniques and technologies to your work.

What we offer

Amazon provides a competitive salary range of $193,300.00 - $261,500.00 USD annually, along with comprehensive benefits including health insurance, retirement plans, and paid time off. You will have the opportunity to work in a dynamic environment that encourages innovation and professional growth. Join us at Amazon and be part of a team that is shaping the future of technology and impacting millions of lives worldwide.