Software Dev Engineer - Compiler, Annapurna Labs

at  Amazon Development Centre Canada ULC

• Currently enrolled, or completed within the last 12 months, a Bachelor’s degree program or higher Computer Science, Computer Engineering, Electrical Engineering, in these fields are considered with a graduation conferral date between December 2024 and September 2025
• Programming experience in internship or coursework with programming language such as Python, C, or C++.

Candidates with strong interests and academic qualifications/research focus in two of the following:

• Data structures and algorithms
• Compilers and Instruction Set Architecture

Knowledge of resource management, scheduling, code generation, and compute graph optimization
Exposure to front-end ML frameworks (i.e. PyTorch, TensorFlow, etc.)
At AWS our vision is to make deep learning pervasive for everyday developers and to democratize access to cutting edge infrastructure. In order to deliver on that vision, we’ve created innovative software and hardware solutions that make it possible.
AWS Neuron is the SDK that optimizes the performance of complex neural net models executed on AWS Inferentia and Trainium, our custom chips designed to accelerate deep-learning workloads.
The Neuron SDK consists of a compiler, run-time, and debugger, integrated with Tensorflow, PyTorch, and MXNet. It is preinstalled in AWS Deep Learning AMIs and Deep Learning Containers for customers to quickly get started with running high performance and cost-effective inference.
The Neuron team is looking for students interested in learning about and contributing to deep learning compilers to solve our customers toughest problems. As a compiler engineer on the Neuron team, you will be supporting the ground-up development of a compiler targeting AWS Inferentia and Trainum. This is an opportunity to work on a brand new product where you will architect and implement business-critical features, and be mentored by a brilliant team of experienced engineers.

Key job responsibilities

• Innovating and delivering creative SW Designs to develop new services, solve operational problems, drive improvements in developer velocity, or positively impact operational safety
• Writing requirements capturing documents, design documents, integration test plans, and deployment plans
• Communicating status and progress of deliverables to schedule, and sharing learnings/ innovations with your team and stakeholders

We are open to hiring candidates to work out of one of the following locations:

Toronto, ON, CAN

• Previous technical internship(s), if applicable.
• Experience in automating
• Ability to effectively articulate technical challenges and solutions.

Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer and does not discriminate on the basis of race, national origin, gender, gender identity, sexual orientation, disability, age, or other legally protected status. If you would like to request an accommodation, please notify your Recruiter

• Innovating and delivering creative SW Designs to develop new services, solve operational problems, drive improvements in developer velocity, or positively impact operational safety
• Writing requirements capturing documents, design documents, integration test plans, and deployment plans
• Communicating status and progress of deliverables to schedule, and sharing learnings/ innovations with your team and stakeholder