Current jobs related to ML Compiler Stack Engineer - Old Toronto - Cerebras Systems


  • Old Toronto, Canada Amazon Full time

    Software Development Manager - ML Compiler, AWS Neuron, Annapurna LabsThe Product: AWS Machine Learning accelerators are at the forefront of AWS innovation. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in the cloud. Trainium will deliver the best-in-class ML training performance with the most teraflops (TFLOPS) of...


  • Old Toronto, Canada Amazon Full time

    Software Development Manager - ML Compiler, AWS Neuron, Annapurna LabsThe Product: AWS Machine Learning accelerators are at the forefront of AWS innovation. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in the cloud. Trainium will deliver the best-in-class ML training performance with the most teraflops (TFLOPS) of...


  • Old Toronto, Canada Amazon Full time

    Neuron Compiler Software Engineer, Annapurna MLAt AWS, our vision is to make deep learning pervasive for everyday developers and to democratize access to cutting-edge infrastructure. To deliver on that vision, we’ve created innovative software and hardware solutions that make it possible.AWS Neuron is the SDK that optimizes the performance of complex...


  • Old Toronto, Canada Meta Full time

    In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectures, combined with...


  • Old Toronto, Canada Amazon Full time

    Neuron Compiler Software Engineer, Annapurna MLAt 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, we’ve created innovative software and hardware solutions that make it possible.AWS Neuron is the SDK that optimizes the performance of...


  • Old Toronto, Ontario, Canada Amazon Full time

    About the RoleWe are seeking a highly skilled Senior ML Compiler Engineer to join our team at Amazon Web Services (AWS). As a key member of our team, you will be responsible for designing, implementing, and maintaining complex software solutions that transform service performance, durability, cost, and security.Key ResponsibilitiesDesign and implement...


  • Old Toronto, Canada Amazon Full time

    Job Title: Neuron Compiler Software Engineer, Annapurna MLAbout the Role:AWS Neuron is a software development kit (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.Key Responsibilities:Support the development of a compiler targeting AWS...


  • 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 trusted AWS Neuron Manager, you will be responsible for innovating on behalf of our customers, leveraging your vision and technical communication skills to...


  • Old Toronto, Ontario, Canada Amazon Full time

    About the RoleThe AWS Neuron team is seeking a highly skilled and experienced Manager III to lead a team of compiler engineers in developing, deploying, and scaling a compiler targeting AWS Inferentia and Trainium. As a trusted AWS Neuron Manager, you will be responsible for innovating on behalf of our customers, leveraging your vision and technical...


  • Old Toronto, Canada Amazon Full time

    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, we’ve 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...


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


  • Old Toronto, Canada Meta Full time

    In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectures, combined with...


  • Old Toronto, Canada Meta Full time

    In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new DL/ML model architectures, combined with...


  • Old Toronto, Canada Amazon Full time

    Do you love decomposing problems to develop products that impact millions of people around the world? Would you enjoy identifying, defining, and building software solutions that revolutionize how businesses operate?The Annapurna Labs team at Amazon Web Services (AWS) is looking for a Senior Software Development Engineer to build, deliver, and maintain...


  • Old Toronto, Canada Amazon Full time

    Do you love decomposing problems to develop products that impact millions of people around the world? Would you enjoy identifying, defining, and building software solutions that revolutionize how businesses operate?The Annapurna Labs team at Amazon Web Services (AWS) is looking for a Senior Software Development Engineer to build, deliver, and maintain...


  • Old Toronto, Canada Amazon Full time

    Sr. Software Development Manager - Compiler, AWS Neuron, Annapurna LabsThe Product: AWS Machine Learning accelerators are at the forefront of AWS innovation. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in the cloud. Trainium will deliver the best-in-class ML training performance with the most teraflops (TFLOPS) of...


  • Old Toronto, Canada Amazon Full time

    Sr. Software Development Manager - Compiler, AWS Neuron, Annapurna LabsThe Product: AWS Machine Learning accelerators are at the forefront of AWS innovation. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in the cloud. Trainium will deliver the best-in-class ML training performance with the most teraflops (TFLOPS) of...


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


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


  • Old Toronto, Canada Amazon Full time

    Sr. Software Development Manager - Compiler, AWS Neuron, Annapurna LabsThe Product: AWS Machine Learning accelerators are at the forefront of AWS innovation. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in the cloud. Trainium will deliver the best-in-class ML training performance with the most teraflops (TFLOPS) of...

ML Compiler Stack Engineer

2 months ago


Old Toronto, Canada Cerebras Systems Full time

Cerebras Systems builds the world’s largest AI chip, 56 times larger than GPUs. Our novel wafer-scale architecture provides the AI compute power of dozens of GPUs on a single chip, with the programming simplicity of a single device. This approach allows Cerebras to deliver industry-leading training and inference speeds and empowers machine learning users to effortlessly run large-scale ML applications, without the hassle of managing hundreds of GPUs or TPUs. Cerebras’ current customers include national labs, global corporations across multiple industries, and top-tier healthcare systems. In January, we announced a multi-year, multi-million-dollar partnership with Mayo Clinic, underscoring our commitment to transforming AI applications across various fields.

Job Overview:

We are seeking a highly skilled Compiler Engineer with a passion for optimizing compiler technologies for AI workloads. You will be an integral part of our software compiler stack team, focusing on enhancing our compiler to fully leverage the unique capabilities of our CS3 system. Your work will play a critical role in achieving unprecedented levels of performance, efficiency, and scalability for AI applications.

Key Responsibilities:
  • Design, develop, and optimize compiler technologies for AI chips using LLVM and MLIR frameworks.
  • Identify and address performance bottlenecks, ensuring optimal resource utilization and execution efficiency.
  • Work with the machine learning team to integrate compiler optimizations with AI frameworks and applications.
  • Contribute to the advancement of compiler technologies by exploring new ideas and approaches.
Qualifications:
  • Bachelor’s, Master’s, or Ph.D. in Computer Science, Electrical Engineering, or a related field.
  • Proven experience in compiler development, particularly with LLVM and/or MLIR.
  • Strong background in optimization techniques, particularly those involving NP-hard problems.
  • Proficiency in C/C++ programming and experience with low-level optimization.
  • Familiarity with AI workloads and architectures is a plus.
  • Excellent problem-solving skills and a strong analytical mindset.
  • Ability to work in a fast-paced, collaborative environment.
What We Offer:
  • Competitive salary and benefits package.
  • Opportunities for professional growth and career advancement.
  • A dynamic and innovative work environment.
  • The chance to work on cutting-edge technologies and make a significant impact on the future of AI.

Cerebras Systems is committed to creating an equal and diverse environment and is proud to be an equal opportunity employer. We celebrate different backgrounds, perspectives, and skills. We believe inclusive teams build better products and companies. We try every day to build a work environment that empowers people to do their best work through continuous learning, growth and support of those around them.

#J-18808-Ljbffr