Software Dev Engineer - AI/ML, AWS Neuron Distributed Training

Position Overview

Are you excited about Machine Learning, chip acceleration, compilers, storage, systems or EC2? At Annapurna – the "secret sauce" behind AWS's success – you’ll join a dynamic team of Software and Hardware engineers working on innovative cloud solutions. As an SDE ML Apps Engineer, you will collaborate with Research Engineers and Applied Scientists to develop backend science components, including deep learning models that empower our platform to help non-tech-savvy customers tackle computer vision challenges.

Key Responsibilities

• Innovate and deliver creative software designs to develop new services, solve operational problems, boost developer velocity, and enhance operational safety.
• Prepare requirements capturing documents, design documents, integration test plans, and deployment plans.
• Communicate the status and progress of deliverables, and share learnings and innovations with your team and stakeholders.

Required Qualifications

• Currently enrolled in, or completed, a Bachelor’s degree or higher in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
• Earned a Bachelor’s or Master’s degree between April 2022 and September 2024. Possible start dates for this role are between March 2024 and October 2024.
• Programming experience through internships or coursework using Python and/or C/C++.
• At least 1 year of internship or coursework experience in deep learning and transformer architectures.

Preferred Qualifications

• Previous software engineering experience (internship/professional) with PyTorch, Jax, or TensorFlow, distributed libraries/frameworks, end-to-end model training, and sharding – with opportunities to optimize and scale large deep learning models on Trainium (AWS Machine Learning acceleration) architecture.
• Experience with distributed, multi-tiered systems, algorithms, and relational databases.
• Expertise in optimization mathematics, including linear programming and nonlinear optimization.

Benefits & Perks

• Compensation: Base pay ranges from $99,500/year in the lowest geographic market up to $200,000/year in the highest, with pay determined by market location, job-related skills, and experience.
• Benefits: Total compensation may include equity, sign-on payments, and a comprehensive package of medical, financial, and other benefits.
• For additional details, please visit our benefits page.