Amazon Neuron Compiler Software Engineer Annapurna ML 

Canada, Ontario, Old Toronto 

994917029

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 of May 2025 or later
- 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 3-month internship starting January 2025

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