Neuron Compiler Software Engineer

2 weeks ago


Toronto, Ontario, Canada Amazon Full time
About the Role

We are seeking a talented software engineer to join our Neuron team at Amazon, where you will play a key role in developing a compiler targeting AWS Inferentia and Trainium, our custom Silicon ML accelerators. As a compiler engineer, you will be responsible for implementing business-critical features and working closely with a brilliant team of experienced engineers.

Key Responsibilities
  • Design and develop a compiler that optimizes the performance of complex neural net models executed on AWS Inferentia and Trainium
  • Collaborate with the Neuron team to integrate the compiler with Tensorflow, PyTorch, and MXNet
  • Work on optimizing compilers, code generation, and compute graph optimization
  • Develop and maintain high-quality code, following best practices and Amazon's coding standards
Requirements
  • Experience with at least one modern language such as Java, Python, C++, or C#
  • Currently enrolled in a Bachelor's degree program or higher majoring in Computer Science, Computer Engineering, Electrical Engineering, or related technical field
  • Strong interests and academic qualifications/research focus in two of the following areas: knowledge of code generation, compute graph optimization, resource scheduling, data structure and algorithms, compiler optimization, machine learning, or machine learning frameworks
  • Available for a 3-month internship
Preferred Qualifications
  • Previous technical internship(s)
  • Knowledge of code generation, compute graph optimization, resource scheduling
  • Experience in optimization mathematics such as linear programming and nonlinear optimization
About Amazon

Amazon is committed to a diverse and inclusive workplace. We are an equal opportunity employer and do 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.



  • Old Toronto, Ontario, Canada Amazon Full time

    Neuron Compiler Software EngineerAmazon is seeking a talented Compiler Software Engineer to join the AWS Neuron team. As a member of this team, you will be responsible for developing and maintaining the Neuron compiler, ensuring optimal performance and efficiency. The Neuron SDK is a comprehensive solution that includes a compiler, run-time, and debugger,...


  • Old Toronto, Ontario, Canada Amazon Full time

    Neuron Compiler Software EngineerAmazon is seeking a talented Compiler Software Engineer to join the AWS Neuron team. As a member of this team, you will be responsible for developing and maintaining the Neuron compiler, ensuring optimal performance and efficiency. The Neuron SDK is a comprehensive solution that includes a compiler, run-time, and debugger,...


  • 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...


  • 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...


  • Old Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a talented Compiler Software Engineer to join our AWS Neuron team. As a key member of our team, you will be responsible for developing and optimizing the Neuron compiler, which is a critical component of our deep learning infrastructure.Key ResponsibilitiesDesign and implement compiler optimizations to improve the performance of...


  • Old Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a talented Compiler Software Engineer to join our AWS Neuron team. As a key member of our team, you will be responsible for developing and optimizing the Neuron compiler, which is a critical component of our deep learning infrastructure.Key ResponsibilitiesDesign and implement compiler optimizations to improve the performance of...


  • 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...


  • 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...


  • 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 all developers. Our innovative software and hardware solutions are designed to accelerate deep learning workloads, and we're looking for talented engineers to join our Neuron team.About the RoleAs a compiler engineer on our Neuron team, you'll be...


  • 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 all developers. Our innovative software and hardware solutions are designed to accelerate deep learning workloads, and we're looking for talented engineers to join our Neuron team.About the RoleAs a compiler engineer on our Neuron team, you'll be...


  • Toronto, Ontario, Canada Amazon Full time

    Unlock the Power of Deep Learning with AWSAt Amazon, we're committed to making deep learning accessible to developers of all levels. To achieve this vision, we've created 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...


  • Toronto, Ontario, Canada Amazon Full time

    Unlock the Power of Deep Learning with AWSAt Amazon, we're committed to making deep learning accessible to developers of all levels. To achieve this vision, we've created 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...


  • Toronto, Ontario, Canada Amazon Full time

    Unlock the Power of Deep Learning with AWSAt Amazon, we're committed to making deep learning accessible to developers of all levels. To achieve this vision, we've created 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...


  • Toronto, Ontario, Canada Amazon Full time

    Unlock the Power of Deep LearningAt Amazon, we're committed to making deep learning accessible to all developers. To achieve this vision, we've created 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 our custom chips,...


  • Toronto, Ontario, Canada Amazon Full time

    Unlock the Power of Deep LearningAt Amazon, we're committed to making deep learning accessible to all developers. To achieve this vision, we've created 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 our custom chips,...


  • Toronto, Ontario, Canada Amazon Full time

    Unlock the Power of Deep LearningAt Amazon, we're committed to making deep learning accessible to all developers. To achieve this vision, we've created 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 our custom chips,...


  • Old Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly motivated and talented individual to join our Neuron team as a Compiler Software Engineer Intern. As a member of this team, you will be responsible for developing and optimizing the Neuron compiler, which is a critical component of our deep learning infrastructure.Key ResponsibilitiesDesign and implement compiler...


  • Old Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly motivated and talented individual to join our Neuron team as a Compiler Software Engineer Intern. As a member of this team, you will be responsible for developing and optimizing the Neuron compiler, which is a critical component of our deep learning infrastructure.Key ResponsibilitiesDesign and implement compiler...


  • 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 all developers. To achieve this vision, we've created 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...


  • 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 all developers. To achieve this vision, we've created 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...