Amazon Neuron Compiler Software Engineer Intern Annapurna ML 

Canada, Ontario, Old Toronto 

233910323

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.

- Experience with at least one modern language such as Java, Python, C++, or C# including object-oriented design
- Currently enrolled in a Bachelor’s degree program or higher majoring in Computer Science, Computer Engineering, Electrical Engineering or related technical field with anticipated graduation date on or later than May 2026
- Candidates with strong interests and academic qualifications/research focus in two of the following: 1. Knowledge of code generation, compute graph optimization, resource scheduling, 2. Data structure and algorithms, 3. Compiler - Optimizing compilers (internals of LLVM, clang, etc), 4. Machine Learning - Experience with XLA, TVM, MLIR, LLVM, 5. Machine Learning frameworks (PyTorch, JAX) and Machine Learning models
- Available for 12-month internship starting May 2025

- Previous technical internship(s)
- Knowledge code generation, compute graph optimization, resource scheduling
- Experience in optimization mathematics such as linear programming and nonlinear optimization