Neuron Compiler Software Engineer

1 month ago

Old Toronto, Ontario, Canada Amazon Full time
Unlock the Power of Deep Learning with AWS Neuron

At Amazon, we're committed to making deep learning accessible to everyone. Our innovative software and hardware solutions, including AWS Neuron, are designed to accelerate deep-learning workloads and make it possible for developers to build and deploy AI models with ease.

About the Role

We're seeking a talented Compiler Engineer to join our Neuron team. As a key member of our team, you'll be responsible for developing and optimizing the compiler for AWS Inferentia and Trainium, our custom Silicon ML accelerators. This is an exciting opportunity to work on a cutting-edge product and contribute to the development of a compiler that will shape the future of deep learning.

Responsibilities

  1. Design and develop a compiler that optimizes the performance of complex neural net models executed on AWS Inferentia and Trainium.
  2. Collaborate with our team to integrate the compiler with Tensorflow, PyTorch, and MXNet.
  3. Work with our customers to understand their deep learning workloads and develop solutions to optimize their performance.
  4. Contribute to the development of our compiler and run-time, and participate in code reviews to ensure high-quality code.

Requirements

  1. Experience with at least one modern language such as Java, Python, C++, or C# including object-oriented design.
  2. 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.
  3. 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.
  4. Available for 3-month internship starting January 2025.

Preferred Qualifications

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

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.

  • Neuron Compiler Software Engineer

    2 months ago

    Old Toronto, Ontario, Canada Amazon Full time

    Unlock the Power of Deep Learning with AmazonAt Amazon, we're committed to making deep learning accessible to everyone. To achieve this vision, we've developed innovative software and hardware solutions that accelerate deep learning workloads. Our Neuron SDK is a cutting-edge tool that optimizes the performance of complex neural net models executed on AWS...

  • Neuron Compiler Software Engineer

    2 months ago

    Old Toronto, Ontario, Canada Amazon Full time

    Unlock the Power of Deep Learning with AmazonAt Amazon, we're committed to making deep learning accessible to everyone. To achieve this vision, we've developed innovative software and hardware solutions that accelerate deep learning workloads. Our Neuron SDK is a cutting-edge tool that optimizes the performance of complex neural net models executed on AWS...

  • Neuron Compiler Software Engineer

    2 months ago

    Toronto, Ontario, Canada Amazon Full time

    Join Our Team as a Compiler Engineer for AWS NeuronAt Amazon, we're committed to making deep learning accessible to everyone. To achieve this vision, we've developed innovative software and hardware solutions that accelerate deep learning workloads.AWS Neuron is a cutting-edge SDK that optimizes the performance of complex neural net models executed on AWS...

  • Neuron Compiler Software Engineer

    2 months ago

    Toronto, Ontario, Canada Amazon Full time

    Join Our Team as a Compiler Engineer for AWS NeuronAt Amazon, we're committed to making deep learning accessible to everyone. To achieve this vision, we've developed innovative software and hardware solutions that accelerate deep learning workloads.AWS Neuron is a cutting-edge SDK that optimizes the performance of complex neural net models executed on AWS...

  • Neuron Compiler Software Engineer

    3 weeks ago

    Toronto, Ontario, Canada Amazon Full time

    Unlock the Power of Deep Learning with AWS NeuronAt Amazon, we're committed to making deep learning accessible to everyone. Our Neuron team is working on innovative software and hardware solutions to accelerate deep learning workloads. We're looking for talented students to join our team and contribute to the development of a compiler targeting AWS...

  • Neuron Compiler Software Engineer

    1 month ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleAt Amazon, we're committed to making deep learning pervasive for everyday developers and democratizing access to cutting-edge infrastructure. To deliver on this vision, we've created innovative software and hardware solutions that make it possible.Job SummaryWe're looking for a talented Compiler Engineer to join our Neuron team, where you'll be...

  • Neuron Compiler Software Engineer

    1 month ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleAt Amazon, we're committed to making deep learning pervasive for everyday developers and democratizing access to cutting-edge infrastructure. To deliver on this vision, we've created innovative software and hardware solutions that make it possible.Job SummaryWe're looking for a talented Compiler Engineer to join our Neuron team, where you'll be...

  • Neuron Compiler Software Engineer

    2 months ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a talented software engineer to join our Neuron team, where you will play a key role in developing a compiler targeting AWS Inferentia and Trainium, our custom Silicon ML accelerators.Key ResponsibilitiesDesign and implement business-critical features for our compiler, working closely with a brilliant team of experienced...

  • Neuron Compiler Software Engineer

    2 months ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a talented software engineer to join our Neuron team, where you will play a key role in developing a compiler targeting AWS Inferentia and Trainium, our custom Silicon ML accelerators.Key ResponsibilitiesDesign and implement business-critical features for our compiler, working closely with a brilliant team of experienced...

  • Senior Compiler Engineer

    4 weeks ago

    Old Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Compiler Engineer to join our AWS Neuron team. As a key member of our team, you will be responsible for designing and developing a compiler targeting AWS Inferentia and Trainium, our custom chips designed to accelerate deep-learning workloads.As a Senior Compiler Engineer, you will work closely with our...

  • Senior Compiler Engineer

    4 weeks ago

    Old Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Compiler Engineer to join our AWS Neuron team. As a key member of our team, you will be responsible for designing and developing a compiler targeting AWS Inferentia and Trainium, our custom chips designed to accelerate deep-learning workloads.As a Senior Compiler Engineer, you will work closely with our...

  • Senior Compiler Engineer

    2 months ago

    Old Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Compiler Engineer to join our AWS Neuron team. As a key member of our team, you will be responsible for designing, implementing, and maintaining cutting-edge compiler technology that optimizes the performance of complex neural net models executed on AWS Inferentia and Trainium.Key ResponsibilitiesDesign...

  • Senior Compiler Engineer

    2 months ago

    Old Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Compiler Engineer to join our AWS Neuron team. As a key member of our team, you will be responsible for designing, implementing, and maintaining cutting-edge compiler technology that optimizes the performance of complex neural net models executed on AWS Inferentia and Trainium.Key ResponsibilitiesDesign...

  • Senior Compiler Engineer Manager

    2 weeks ago

    Old Toronto, Ontario, Canada Amazon Full time

    About the RoleThe AWS Neuron team is responsible for building innovative software solutions for AWS customers. As a Manager III on this team, you will lead a team of compiler engineers in developing, deploying, and scaling a compiler targeting AWS Inferentia and Trainium. You will need to be technically capable, credible, and curious in your own right as a...

  • Senior Compiler Engineer

    1 month ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Compiler Engineer to join our AWS Neuron team. As a key member of our team, you will be responsible for designing, developing, and maintaining high-performance compilers for machine learning workloads on AWS Inferentia and Trainium.Key ResponsibilitiesDesign and develop innovative compiler solutions to...

  • Senior Compiler Engineer

    1 month ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly skilled Senior Compiler Engineer to join our AWS Neuron team. As a key member of our team, you will be responsible for designing, developing, and maintaining high-performance compilers for machine learning workloads on AWS Inferentia and Trainium.Key ResponsibilitiesDesign and develop innovative compiler solutions to...

  • Neural Network Compiler Software Engineer

    2 months ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleAt Amazon, we're committed to making deep learning pervasive for everyday developers and democratizing access to cutting-edge infrastructure. To achieve this vision, we've created innovative software and hardware solutions that make it possible.AWS Neuron: Optimizing Deep Learning WorkloadsAWS Neuron is the SDK that optimizes the performance of...

  • Neural Network Compiler Software Engineer

    2 months ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleAt Amazon, we're committed to making deep learning pervasive for everyday developers and democratizing access to cutting-edge infrastructure. To achieve this vision, we've created innovative software and hardware solutions that make it possible.AWS Neuron: Optimizing Deep Learning WorkloadsAWS Neuron is the SDK that optimizes the performance of...

  • Software Compiler Manager

    4 weeks ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly skilled and experienced Manager III to lead our team of compiler engineers in developing, deploying, and scaling compilers for AWS Inferentia and Trainium. As a key member of the AWS Neuron team, you will collaborate with AWS ML services teams to innovate and ensure the Neuron SDK meets customer needs for high...

  • Software Compiler Manager

    4 weeks ago

    Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly skilled and experienced Manager III to lead our team of compiler engineers in developing, deploying, and scaling compilers for AWS Inferentia and Trainium. As a key member of the AWS Neuron team, you will collaborate with AWS ML services teams to innovate and ensure the Neuron SDK meets customer needs for high...