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

United States, California, Cupertino 

11610594

As an SDE ML Apps Engineer, you will work alongside Research Engineers and Applied Scientists to build backend science components, including deep learning models that power our platform. Our platform enables non-tech-savvy customers to understand and solve their computer vision
problems.Key job responsibilities- Writing requirements capturing documents, design documents, integration test plans, and deployment plans

- Currently enrolled in, or completed a Bachelor’s degree program or higher in Computer Science, Computer Engineering, Electrical Engineering or related field
- To qualify, applicants should have earned a Bachelor’s or Master’s degree between April 2022 to September 2024. Possible start dates for this role are between March 2024 to October 2024.
- Programming experience in internship or coursework with programming language such as Python and/or C or C++.
- 1+ years of internship or coursework in deep learning, transformer architectures

• Previous software engineer (internship/professional) experience with Pytorch/Jax/Tensorflow, Distributed libraries and Frameworks, end-to-end model training, and sharding. The group presents lot of opportunity for optimization and scaling large deep learning models on Trainium (AWS Machine Learning acceleration) architecture.
• Experience with distributed, multi-tiered systems, algorithms, and relational databases.
• Experience in optimization mathematics such as linear programming and nonlinear optimization