ML Compiler Engineer

3 weeks ago


Vancouver, 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 Software Development Engineer II to build, deliver, and maintain complex products that delight our customers and raise our performance bar. You’ll design fault-tolerant systems that run at massive scale as we continue to innovate best-in-class services and applications in the AWS Cloud.

Annapurna Labs was a startup company acquired by AWS in 2015, and is now fully integrated. If AWS is an infrastructure company, then think Annapurna Labs as the infrastructure provider of AWS. Our org covers multiple disciplines including silicon engineering, hardware design and verification, software, and operations. AWS Nitro, ENA, EFA, Graviton and F1 EC2 Instances, AWS Neuron, Inferentia and Trainium ML Accelerators, and in storage with scalable NVMe, are some of the products we have delivered, over the last few years.

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 and Trainium, our custom chips designed to accelerate deep-learning workloads.

The Neuron SDK consists of a compiler, run-time, and debugger, integrated with Tensorflow, PyTorch, and MXNet. It’s preinstalled in AWS Deep Learning AMIs and Deep Learning Containers for customers to quickly get started with running high performance and cost-effective inference.

The Neuron team is hiring compiler engineers in order to solve our customers toughest problems.

This is an opportunity to work on cutting-edge products at the intersection of machine-learning, high-performance computing, and distributed architectures. You will architect and implement business-critical features, publish cutting-edge research, and mentor a brilliant team of experienced engineers. We operate in spaces that are very large, yet our teams remain small and agile. There is no blueprint. We're inventing. We're experimenting. It is a very unique learning culture.

As a deep learning compiler engineer on the Neuron team, you will be a thought leader supporting the development of a compiler targeting AWS Inferentia and Trainium. You will be developing and scaling the compiler to handle the world's largest ML workloads. You will need to be technically capable, credible and curious in your own right as a trusted AWS Neuron engineer, innovating on behalf of our customers. You will leverage your technical communications skill as a hands-on partner to AWS ML services teams and you will be involved in pre-silicon design, bringing new products/features to market, and many other exciting projects. A background in machine learning and AI accelerators is preferred, but not required.

Key job responsibilities:

  1. Solve challenging technical problems, often ones not solved before, at every layer of the stack.
  2. Design, implement, test, deploy and maintain innovative software solutions to transform service performance, durability, cost, and security.
  3. Build high-quality, highly available, always-on products.
  4. Research implementations that deliver the best possible experiences for customers.

A day in the life:

  1. As you design and code solutions to help our team drive efficiencies in software architecture, you’ll create metrics, implement automation and other improvements, and resolve the root cause of software defects.
  2. Build high-impact solutions to deliver to our large customer base.
  3. Participate in design discussions, code review, and communicate with internal and external stakeholders.
  4. Work cross-functionally to help drive business decisions with your technical input.
  5. Work in a startup-like development environment, where you’re always working on the most important stuff.

About the team:

  1. Diverse Experiences: AWS values diverse experiences. Even if you do not meet all of the qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.
  2. Why AWS: Amazon Web Services (AWS) is the world’s most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating — that’s why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses.
  3. Inclusive Team Culture: Here at AWS, we embrace our differences. We are committed to furthering our culture of inclusion. We have ten employee-led affinity groups, reaching 40,000 employees in over 190 chapters globally. We have innovative benefit offerings, and host annual and ongoing learning experiences, including our Conversations on Race and Ethnicity (CORE) and AmazeCon (gender diversity) conferences.
  4. Work/Life Balance: Our team puts a high value on work-life balance. It isn’t about how many hours you spend at home or at work; it’s about the flow you establish that brings energy to both parts of your life. We believe striking the right balance between your personal and professional life is critical to life-long happiness and fulfillment. We offer flexibility in working hours and encourage you to find your own balance between your work and personal lives.
  5. Mentorship & Career Growth: Our team is dedicated to supporting new members. We have a broad mix of experience levels and tenures, and we’re building an environment that celebrates knowledge sharing and mentorship. We care about your career growth and strive to assign projects based on what will help each team member develop into a better-rounded professional and enable them to take on more complex tasks in the future.

Minimum Qualifications:

  • 3+ years of non-internship professional software development experience.
  • 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience.
  • Experience programming with at least one software programming language.
  • 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience.
  • Bachelor's degree in computer science or equivalent.

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.

#J-18808-Ljbffr
  • ML Compiler Engineer

    2 months ago


    Vancouver, 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 Software Development Engineer II to build, deliver, and maintain complex...


  • Vancouver, British Columbia, Canada Square Full time

    At Square, we're building a platform that protects our customers and our business from fraud and payment disputes. Our Risk ML Platform is a critical part of this effort, and we're seeking an experienced software engineer to join our team.What You'll DoYou'll work across the entire machine learning lifecycle, from training to deployment and model...


  • Vancouver, British Columbia, Canada Microsoft Full time

    Job OverviewWe are seeking a highly skilled Senior GPU Compiler Engineer to join our team at Microsoft, working on the development of GPU programming via HLSL. This role will involve working with industry partners to ensure our compiler delivers best-in-class content and performance, as well as collaborating with customers, partners, and stakeholders to...


  • Vancouver, Canada Huawei Technologies Canada Co., Ltd. Full time

    Our team has an immediate permanent opening for an Engineer. Responsibilities: Apply relevant AI infrastructure and software/hardware acceleration techniques to build & optimize our intelligent AI/ML systems that improve our products and experiences.Apply your distributed system experience to build & optimize the AI/ML infrastructure for scalability,...


  • Vancouver, Canada Amgen Full time

    Join Amgen’s Mission of Serving Patients At Amgen, if you feel like you’re part of something bigger, it’s because you are. Our shared mission—to serve patients living with serious illnesses—drives all that we do. Since 1980, we’ve helped pioneer the world of biotech in our fight against the world’s toughest diseases. With our focus on four...

  • AI Compiler Architect

    3 weeks ago


    Vancouver, British Columbia, Canada Microsoft Canada Full time

    Job SummaryWe are seeking a highly skilled AI Compiler Architect to join our team at Microsoft Canada. This is an exciting opportunity to work with cutting-edge technology and collaborate with inter-disciplinary teams.About the RoleIn this role, you will be responsible for designing and building AI compilers, developing new AI programming abstractions, and...

  • AI/ML Technical Lead

    4 weeks ago


    Vancouver, Canada ITC Infotech Full time

    Greetings from ITC Infotech! Job Title - AI/ML Technical LeadLocation – Seattle, WA/ Vancouver, CAJob Description:Experience in building models and extracting insights with supervised and unsupervised techniques using both machine learning and deep learning methodologies.Ability to work with choice of hyperparameters for optimizing performance towards...


  • Vancouver, Canada Amazon Development Centre Canada ULC - K03 Full time

    PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience - Experience in patents or publications at top-tier peer-reviewed conferences or journals - Experience programming in Java, C++, Python or related language - Experience in any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining,...

  • AI/ML Technical Lead

    2 weeks ago


    Vancouver, British Columbia, Canada Electronic Arts Full time

    Job SummaryWe are seeking a highly experienced AI/ML Technical Lead to join our team at Electronic Arts. This role will lead and manage a team of Machine Learning Engineers to develop and implement AI/ML algorithms for in-game recommendation and matchmaking systems.About the RoleThis is a leadership position that requires strong technical expertise and...

  • AI/ML Technical Lead

    2 weeks ago


    Vancouver, British Columbia, Canada Electronic Arts Full time

    Job OverviewWe are seeking an experienced AI/ML Technical Lead to join our team at Electronic Arts. In this role, you will lead and manage a team of Machine Learning Engineers to develop and implement AI/ML algorithms for in-game recommendation and matchmaking systems.Key ResponsibilitiesDrive the development and execution of AI/ML strategies to enhance game...

  • AI/ML Technical Lead

    1 month ago


    Vancouver, British Columbia, British Columbia, Canada ITC Infotech Full time

    Greetings from ITC Infotech! Job Title - AI/ML Technical LeadLocation – Seattle, WA/ Vancouver, CAJob Description:Experience in building models and extracting insights with supervised and unsupervised techniques using both machine learning and deep learning methodologies.Ability to work with choice of hyperparameters for optimizing performance towards...


  • Vancouver, British Columbia, Canada Huawei Technologies Canada Co., Ltd. Full time

    Our team at Huawei Technologies Canada Co., Ltd. has an exciting opportunity for a highly skilled AI Infrastructure Software Engineer.Responsibilities:Develop and optimize AI infrastructure and software/hardware acceleration techniques to improve our intelligent AI/ML systems.Apply distributed system experience to build and optimize AI/ML infrastructure for...


  • Vancouver, Canada Fynbosys Inc Full time

    **Profile** **Must haves**: - Acknowledges the presence of choice in every moment and takes personal responsibility for their life. - Possesses an entrepreneurial spirit and continuously innovates to achieve great results. - Communicates with honesty and kindness and creates the space for others to do the same. - Leads with courage, knowing the possibility...

  • DataOps Engineer

    1 month ago


    Vancouver, Canada T-Net British Columbia Full time

    Sanctuary - a multi award-winning LinkedIn Top Startup company - is looking to hire a DataOps Engineer for our Data Collection team. Reporting to the Data Collection Team Lead, you'll gain a comprehensive understanding of the data collection tools and ML training pipelines that power our general Purpose Robots (GPRs).The best candidate for this role will be...


  • Vancouver, British Columbia, Canada Flow Full time

    Flow Blockchain is seeking a talented Compiler Developer to join its team.As a key member of our engineering team, you will be responsible for designing and implementing compilers to support the development of smart contract programming languages.The ideal candidate will have practical experience with compiler design and implementation, as well as a strong...


  • Vancouver, British Columbia, Canada T-Net British Columbia Full time

    Transform Robot Telemetry into ML Training DatasetsJob SummaryWe're seeking a skilled DataOps Engineer to join our Data Collection team at T-Net British Columbia. As a DataOps Engineer, you'll gain a comprehensive understanding of our data collection tools and ML training pipelines that power our General Purpose Robots.The ideal candidate will have proven...

  • Senior Data Scientist

    4 weeks ago


    Vancouver, British Columbia, Canada Global Relay Full time

    About Global Relay:At Global Relay, we're a leader in enterprise information archiving, offering cloud archiving, surveillance, eDiscovery, and analytics solutions. Our mission is to help highly regulated firms securely capture and preserve their communications data, ensuring compliance with stringent regulations.Your Role:As a Senior Data Scientist, you'll...


  • Vancouver, British Columbia, Canada Canada Workday ULC Full time

    About the RoleAs a key member of our Machine Learning team, you will play a pivotal role in driving the development of our AI/ML capabilities within the Prism platform.About YouYou have a passion for working on complex technical projects and are an authority on modern ML/AI technologies.You possess excellent leadership skills and can lead end-to-end ML/AI...


  • Vancouver, British Columbia, Canada Canada Workday ULC Full time

    About the RoleAs a key member of our Machine Learning team, you will play a pivotal role in driving the success of our Prism platform. You will collaborate closely with product managers, designers, architects, and engineers to design, develop, and deploy cutting-edge ML/AI capabilities.Key Responsibilities:You will lead the development of sophisticated ML/AI...

  • DataOps Engineer

    1 month ago


    Vancouver, Canada Sanctuary Cognitive Systems Corp. Full time

    Your New Role and Team Sanctuary - a multi award-winning LinkedIn Top Startup company - is looking to hire a DataOps Engineer for our Data Collection team. Reporting to the Data Collection Team Lead, you’ll gain a comprehensive understanding of the data collection tools and ML training pipelines that power our general Purpose Robots (GPRs).The best...