Talent.com
PyTorch Operator - ML Engineer
PyTorch Operator - ML EngineerMercor • Hampton, Virginia, US
PyTorch Operator - ML Engineer

PyTorch Operator - ML Engineer

Mercor • Hampton, Virginia, US
23 hours ago
Job type
  • Full-time
  • Remote
Job description

###

  • 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
  • Create a job alert for this search

    Ml Engineer • Hampton, Virginia, US

    Related jobs
    Toolmaker / CNC Programmer

    Toolmaker / CNC Programmer

    Howmet • Hampton, VA, United States
    Full-time
    One Howmet Drive, Hampton, VA, 23661, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability...Show more
    Last updated: 30+ days ago • Promoted
    Facilities Project Engineer

    Facilities Project Engineer

    Howmet • Hampton, VA, United States
    Full-time
    One Howmet Drive, Hampton, VA, 23661, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability...Show more
    Last updated: 30+ days ago • Promoted
    Team Leader - Segment Work Cells - First Shift

    Team Leader - Segment Work Cells - First Shift

    Howmet • Hampton, VA, United States
    Full-time
    One Howmet Drive, Hampton, VA, 23661, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability...Show more
    Last updated: 21 days ago • Promoted
    MGR ENGINEERING 2

    MGR ENGINEERING 2

    HII • Newport News, VA, US
    Full-time
    Entity : Newport News Shipbuilding.US Citizenship Required for this Position : Yes.Relocation : No relocation assistance available. Virtual / Telework Opportunity : No - Not eligible for telework.Travel R...Show more
    Last updated: 30+ days ago • Promoted
    Trellix Endpoint Engineer

    Trellix Endpoint Engineer

    Booz Allen Hamilton • Norfolk, VA, United States
    Full-time +1
    Design, deploy, manage, and operationalize enterprise endpoint data protection controls as a Trellix Endpoint Data Loss Prevention (DLP) Engineer. Serve as the technical owner for Trellix EDR / DLP co...Show more
    Last updated: 6 days ago • Promoted
    Senior CMM Programmer

    Senior CMM Programmer

    Howmet • Hampton, VA, United States
    Full-time
    One Howmet Drive, Hampton, VA, 23661, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability...Show more
    Last updated: 30+ days ago • Promoted
    Project Engineer (Mechanical)

    Project Engineer (Mechanical)

    Oceaneering International, Inc. • Chesapeake, VA, United States
    Full-time
    Oceaneering Marine Services Division (MSD) has over 30 years of experience providing full-service submarine and surface ship repairs supporting maintenance and alterations aboard commercial and U.W...Show more
    Last updated: 30+ days ago • Promoted
    Data Engineer (Navy / DoD)

    Data Engineer (Navy / DoD)

    THOR Solutions, LLC • Norfolk, VA, US
    Temporary
    US Navy Ship Anti-Submarine Warfare (ASW) Readiness and Effectiveness Measuring (SHAREM) at Surface and Mine Warfighting Development Center (SMWDC) in. This position is associated with an upcoming c...Show more
    Last updated: 30+ days ago • Promoted
    Software Tool Manager

    Software Tool Manager

    Leidos Inc • Newport News, VA, United States
    Full-time
    ACCA5JI Joint Interoperability Tactical Command and Control (JINTACCS) mission at the customer office in Langley, VA.In this role, you will leverage your expertise to manage and deploy the iSMART s...Show more
    Last updated: 30+ days ago • Promoted
    Product Engineer Team Leader - HSC

    Product Engineer Team Leader - HSC

    Howmet • Hampton, VA, United States
    Full-time
    One Howmet Drive, Hampton, VA, 23661, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability...Show more
    Last updated: 30+ days ago • Promoted
    Product Engineer

    Product Engineer

    Howmet • Hampton, VA, United States
    Full-time
    One Howmet Drive, Hampton, VA, 23661, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability...Show more
    Last updated: 30+ days ago • Promoted
    Workday HCM Integration Lead and Developer

    Workday HCM Integration Lead and Developer

    My3Tech Inc • Hampton, VA, United States
    Full-time
    Workday HCM Integration Lead / Developer.The Workday Integration Lead / Developer primary responsibility will be to provide lead support, development, testing and management of integrations of our Work...Show more
    Last updated: 21 days ago • Promoted
    MGR ENGINEERING PLANNING 2

    MGR ENGINEERING PLANNING 2

    HII • Newport News, VA, US
    Full-time
    Newport News, Virginia, United States.Entity : Newport News Shipbuilding.US Citizenship Required for this Position : Yes. Relocation : No relocation assistance available.Virtual / Telework Opportunity : N...Show more
    Last updated: 30+ days ago • Promoted
    CMM Programmer

    CMM Programmer

    Howmet • Hampton, VA, United States
    Full-time
    One Howmet Drive, Hampton, VA, 23661, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability...Show more
    Last updated: 30+ days ago • Promoted
    Tool & Wax Die Maintenance Tech

    Tool & Wax Die Maintenance Tech

    Howmet • Hampton, VA, United States
    Full-time
    One Howmet Drive, Hampton, VA, 23661, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability...Show more
    Last updated: 29 days ago • Promoted
    Multicraft Maintenance Tech

    Multicraft Maintenance Tech

    Howmet • Hampton, VA, United States
    Full-time
    One Howmet Drive, Hampton, VA, 23661, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability...Show more
    Last updated: 30+ days ago • Promoted
    Workday HCM Integration Lead and Developer

    Workday HCM Integration Lead and Developer

    Apex Systems • Hampton, VA, United States
    Full-time
    Position : Workday HCM Integration Lead / Developer.The Workday Integration Lead / Developer primary responsibility will be to provide lead support, development, testing and management of integrations o...Show more
    Last updated: 20 days ago • Promoted
    Project Engineer III

    Project Engineer III

    Oceaneering International, Inc. • Chesapeake, VA, United States
    Full-time
    Oceaneering Marine Services Division (MSD) has over 30 years of experience providing full-service submarine and surface ship repairs supporting maintenance and alterations aboard commercial and U.W...Show more
    Last updated: 30+ days ago • Promoted