Current jobs related to PyTorch Operator - Toronto, Ontario - Mercor

  • CUDA Kernel Optimizer

    1 week ago

    Toronto, Ontario, Canada MERCOR Full time $32,000 - $64,000 per year

    1) Role OverviewMercor is engaging advanced CUDA experts who specialize in GPU kernel optimization, performance profiling, and numerical efficiency. These professionals possess a deep mental model of how modern GPU architectures execute deep learning workloads. They are comfortable translating algorithmic concepts into finely tuned kernels that maximize...

  • HPC Engineer, AI/ML Infrastructure

    7 days ago

    Toronto, Ontario, Canada Boson AI Full time $120,000 - $180,000 per year

    About The Role We're looking for a Senior High Performance Computing Engineer to help us run one of the most exciting GPU clusters around—our Toronto datacenter packed with NVIDIA H100 and A100 GPUs, over 20PB of Ceph storage, terabit networking, and hundreds of servers You'll be hands-on with the full lifecycle of HPC infrastructure: planning, building,...

  • Site Reliability Engineer, AI/ML Infrastructure

    2 hours ago

    Toronto, Ontario, Canada Boson AI Full time $120,000 - $180,000 per year

    About The Role We're looking for a Senior Site Reliability Engineer to help us run one of the most exciting GPU clusters around—our Toronto datacenter packed with NVIDIA H100 and A100 GPUs, over 20PB of Ceph storage, terabit networking, and hundreds of servers You'll be hands-on with the full lifecycle of HPC infrastructure: planning, building, testing,...

  • Machine Learning Engineer

    1 week ago

    Toronto, Ontario, Canada Taalas Full time $120,000 - $180,000 per year

    At Taalas we believe that fundamental progress is achieved by those who are willing to understand and assail a problem end-to-end, without regard for commonly accepted abstractions and boundaries.We are building a team of hands-on technologists who dislike overspecialization and seek to excel in both depth and breadth.In this position the successful...

  • Senior Machine Learning Engineer, Large Language Models

    2 weeks ago

    Toronto, Ontario, Canada SoundHound AI Full time $120,000 - $180,000 per year

    The Senior Machine Learning Engineer, Large Language Models begins operating as a senior-level contributor, moving beyond feature ownership to drive small projects end-to-end. They apply deep ML expertise, ensure quality delivery, and start shaping the technical standards of the team while still receiving direction on priorities.What You'll DoDesign and...

  • Lead Engineer, Generative AI and Machine Learning

    3 hours ago

    Toronto, Ontario, Canada RBC Full time $120,000 - $180,000 per year

    Job DescriptionWhat is the opportunity? This role offers a unique chance to pioneer the integration of Generative AI and machine learning (ML) into Site Reliability Engineering (SRE), driving transformative improvements in system reliability, efficiency, and scalability. You will work at the intersection of AI/ML innovation and cloud-native infrastructure,...

  • ML Compiler Engineer

    2 weeks ago

    Toronto, Ontario, Canada Amazon Full time $120,000 - $200,000 per year

    DESCRIPTIONThe Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon's custom machine learning accelerators, Inferentia and Trainium.The Product: The AWS Machine Learning accelerators (Inferentia/Trainium) offer unparalleled ML inference and training...

  • ML Compiler Engineer

    1 week ago

    Toronto, Ontario, Canada Amazon Web Services (AWS) Full time US$140,000 - US$200,000 per year

    DescriptionThe Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon's custom machine learning accelerators, Inferentia and Trainium.The Product: The AWS Machine Learning accelerators (Inferentia/Trainium) offer unparalleled ML inference and training...

  • AI Implementation Lead

    1 week ago

    Toronto, Ontario, Canada ND Paper Full time $120,000 - $180,000 per year

    Job Title: Artificial Intelligence (AI) Implementation LeadAbout UsND Paper is a leading manufacturer of high-quality pulp, paper, and packaging products in the United States, generating over half a billion dollars in annual sales. As a wholly owned subsidiary of Nine Dragons Paper (Holdings) Limited – the largest containerboard producer in the world –...

  • Team Lead AI Engineer

    2 weeks ago

    Toronto, Ontario, Canada Typescouts Full time $54,000 - $72,000 per year

    Salary: USD $4,500 – $6,000 / month PTOContract: Independent Contractor AgreementWorking Hours: 9am – 5pm ESTLocation: Canada (Remote)About the CompanyOur client is a growing Canadian startup that creates innovative tools to simplify and consolidate business operations. They achieve this by integrating applications and data into a single, intelligent...

PyTorch Operator

2 weeks ago

Toronto, Ontario, Canada Mercor Full time US$20,800 - US$104,000 per year

1) Role Overview

Mercor is seeking experienced PyTorch experts who excel in extending and customizing the framework at the operator level. Ideal contributors are those who deeply understand PyTorch's dispatch system, ATen, autograd mechanics, and C++ extension interfaces. These contractors bridge research concepts and high-performance implementation, producing clear, maintainable operator definitions that integrate seamlessly into existing codebases.

2) Key Responsibilities

  • Design and implement new PyTorch operators and tensor functions in C++/ATen.

  • Build and validate Python bindings with correct gradient propagation and test coverage.

  • Create "golden" reference implementations in eager mode for correctness validation.

  • Collaborate asynchronously with CUDA or systems engineers who handle low-level kernel optimization.

  • Profile, benchmark, and report performance trends at the operator and graph level.

  • Document assumptions, APIs, and performance metrics for reproducibility.

3) Ideal Qualifications

  • Deep understanding of PyTorch internals (TensorIterator, dispatcher, autograd engine).

  • Strong background in C++17+ and template metaprogramming within PyTorch's ecosystem.

  • Experience authoring or extending PyTorch custom ops or backends.

  • Working knowledge of performance profiling tools and GPU/CPU interplay.

  • Strong written communication and ability to deliver well-documented, self-contained modules.

  • Prior open-source contributions to PyTorch, TorchInductor, Triton, or related projects are a plus.

4) More About the Opportunity

  • Ideal for contractors who enjoy building clean, high-performance abstractions in deep learning frameworks.

  • Work is asynchronous, flexible, and outcome-oriented.

  • Collaborate with CUDA optimization specialists to integrate and validate kernels.

  • Projects may involve primitives used in state-of-the-art AI models and benchmarks.

5) Compensation & Contract Terms

  • Typical range: $100–$200/hour, depending on experience and project scope.

  • Structured as an independent contractor engagement, not employment.

  • Payments for services rendered on a milestone or weekly invoice cadence.

  • Confidentiality and IP assignment agreements may apply.

6) Application Process

  • Share a concise summary of your experience with PyTorch internals and systems-level programming.

  • Include links to open-source work, GitHub PRs, or sample operator implementations.

  • Provide hourly rate, availability, and relevant technical background.

  • Selected experts may complete a short, paid pilot module to demonstrate fit.

7) About Mercor

  • Mercor connects top-tier independent professionals with leading AI labs and research organizations.

  • Contractors retain full control over their work, schedule, and deliverables.

  • Our engagements focus on high-skill, high-autonomy technical work across AI systems and optimization