ML Compiler Software Engineer PEY Co-op 12-16 months Annapurna ML
ML Compiler Software Engineer PEY Co-op 12-16 months Annapurna ML
Posted on :
24-04-2025
Employer Active
1 Vacancy
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Share
Send me jobs like this
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Job Description
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 weve 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 deeplearning workloads.
The Neuron SDK consists of a compiler runtime 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 costeffective inference and training.
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 development of a compiler targeting AWS Inferentia and Trainium our custom Silicon ML accelerators. This is an opportunity to work on a cuttingedge product where you will implement businesscritical features and be mentored by a brilliant team of experienced engineers.
Experience with at least one modern language such as Java Python C or C# including objectoriented design
Currently enrolled in a Bachelors 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 12month 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
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status disability or other legally protected status.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process including support for the interview or onboarding process please visit for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
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 deeplearning workloads.
The Neuron SDK consists of a compiler runtime 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 costeffective inference and training.
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 development of a compiler targeting AWS Inferentia and Trainium our custom Silicon ML accelerators. This is an opportunity to work on a cuttingedge product where you will implement businesscritical features and be mentored by a brilliant team of experienced engineers.
Experience with at least one modern language such as Java Python C or C# including objectoriented design
Currently enrolled in a Bachelors 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 12month 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
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status disability or other legally protected status.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process including support for the interview or onboarding process please visit for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
Employment Type
Full-Time
Department / Functional Area
Key Skills
Report This Job
Disclaimer: Drjobpro.com is only a platform that connects job seekers and employers. Applicants are advised to conduct their own independent research into the credentials of the prospective employer.We always make certain that our clients do not endorse any request for money payments, thus we advise against sharing any personal or bank-related information with any third party. If you suspect fraud or malpractice, please contact us via contact us page.
