Talent.com
PyTorch Operator - ML Engineer
PyTorch Operator - ML EngineerMercor • Bentonville, Arkansas, US
PyTorch Operator - ML Engineer

PyTorch Operator - ML Engineer

Mercor • Bentonville, Arkansas, US
22 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 • Bentonville, Arkansas, US

    Related jobs
    Software Engineer II

    Software Engineer II

    Walmart • Bentonville, AR, United States
    Full-time +1
    Application & Feature Delivery : .Executes scoped feature or model delivery through collaborative engineering teams.Owns implementation, iteration, testing, and support for assigned services, APIs, u...Show more
    Last updated: 3 days ago • Promoted
    GTW SW Op Excellence

    GTW SW Op Excellence

    My3Tech Inc • Bentonville, AR, United States
    Full-time
    ONSITE IN SUNNYVALE (2 Days per week).INCLUDE WALMART REFS IF APPLICABLE • •.DO NOT USE ANYONE WHO HAS WORKED IN ANY AMOUNT LAYERS WITH THE FOLLOWING : . Identify, develop and execute opportunities to r...Show more
    Last updated: 3 days ago • Promoted
    Software Engineer II - Fullstack / AI

    Software Engineer II - Fullstack / AI

    Lennar • Bentonville, AR, United States
    Full-time
    Lennar is one of the nation's leading homebuilders, dedicated to making an impact and creating an extraordinary experience for their Homeowners, Communities, and Associates by building quality home...Show more
    Last updated: 1 hour ago • Promoted • New!
    Software Engineer III

    Software Engineer III

    Walmart • Bentonville, AR, United States
    Full-time +1
    Our Team : Health and Wellness - Store Technology.As a member of the Health and Wellness Quality Engineering group, you'll be responsible for test designing, test execution, and find defects using b...Show more
    Last updated: 30+ days ago • Promoted
    (USA) Senior, Software Engineer - Unified Payments Platform

    (USA) Senior, Software Engineer - Unified Payments Platform

    Walmart • Bentonville, AR, United States
    Full-time +1
    Walmart's physical and digital channels.The platform is being re-architected from a legacy mainframe to a modern, cloud-native solution. Walmart's Unified Payments Platform.You'll work closely with ...Show more
    Last updated: 30+ days ago • Promoted
    Senior, Software Engineer - Transactional Systems

    Senior, Software Engineer - Transactional Systems

    Walmart • Bentonville, AR, United States
    Full-time +1
    Senior, Software Engineer - Transactional Systems.Yes, we are the Fortune #1 company.But you'll quickly find we're a company who wants you to feel comfortable bringing your whole self to work.A car...Show more
    Last updated: 30+ days ago • Promoted
    Senior, Software Engineer - Full Stack, Indirect Procurement (Evergreen)

    Senior, Software Engineer - Full Stack, Indirect Procurement (Evergreen)

    Walmart • Bentonville, AR, United States
    Full-time
    Imagine working in an environment where one line of code can make life easier for hundreds of millions of people.That's what we do at Walmart Global Tech. We're a team of software engineers, data sc...Show more
    Last updated: 3 days ago • Promoted
    (USA) Software Engineer III (Full Stack)

    (USA) Software Engineer III (Full Stack)

    Walmart • Bentonville, AR, United States
    Full-time +1
    Walmart's Enterprise Business Services (EBS) is a powerhouse of seven exceptional teams delivering world-class technology solutions and services making a profound impact at every level of Walmart.A...Show more
    Last updated: 30+ days ago • Promoted
    (USA) Software Engineer III - Data Foundation Services

    (USA) Software Engineer III - Data Foundation Services

    Walmart • Bentonville, AR, United States
    Full-time +1
    As an engineer on the Data Foundation Services(DFS) team, you will contribute to designing, building , deploying and operating our best of breed database services at scale.We currently operate a la...Show more
    Last updated: 3 days ago • Promoted
    Software Engineer III Fullstack

    Software Engineer III Fullstack

    Walmart • Bentonville, AR, United States
    Full-time +1
    Walmart Job Description : Full Stack Engineer As a Full Stack Engineer at Walmart, you will play a key role within our engineering projects. You will provide technical and code creation, documentatio...Show more
    Last updated: 30+ days ago • Promoted
    Software Engineer III - Tableau Platform

    Software Engineer III - Tableau Platform

    Walmart • Bentonville, AR, United States
    Full-time +1
    We are looking for a Platform Engineer to join our Analytics and Visualization Platform team, focused on enabling analytics and visualization requirements on the Tableau Platform.In this role, you ...Show more
    Last updated: 3 days ago • Promoted
    Software Engineer III - Backend

    Software Engineer III - Backend

    Walmart • Bentonville, AR, United States
    Full-time +1
    Focusing on customer, associate and business needs, this team works with Walmart International, which includes more than 5,200 retail units, operating in 23 countries such as Canada, Central Americ...Show more
    Last updated: 3 days ago • Promoted
    (USA) Software Engineer II

    (USA) Software Engineer II

    Walmart • Bentonville, AR, United States
    Full-time +1
    This is a leadership role that combines technical skills with project management and leadership responsibilities.The ideal candidate will have experience with SQL, software development, and leaders...Show more
    Last updated: 1 day ago • Promoted
    (USA) Software Engineer III - Full Stack, GenAI Platform

    (USA) Software Engineer III - Full Stack, GenAI Platform

    Walmart • Bentonville, AR, United States
    Full-time +1
    Generative AI-powered solutions that improve associate productivity and enable internal teams to create their own AI-driven capabilities. This role provides the opportunity to grow your technical ex...Show more
    Last updated: 3 days ago • Promoted
    Power Equipment Operator

    Power Equipment Operator

    Walmart • Bentonville, AR, US
    Full-time +1
    Career opportunities in Power Equipment Operator roles include Lift Driver, Motorized Pallet Jack, Stock Picker, Clamp Lift and On Property Yard Driver. As a Power Equipment Operator at Walmart Supp...Show more
    Last updated: 30+ days ago • Promoted
    Lab Tech Cutting Operator

    Lab Tech Cutting Operator

    Staffmark Group • Bentonville, AR, United States
    Full-time
    Do you enjoy hands-on work and.We're hiring a Lab Tech Cutting Operator to join the Cutting Department and support product testing operations. If you're dependable, detail-oriented, and comfortable ...Show more
    Last updated: 30+ days ago • Promoted
    Azure Data Engineer

    Azure Data Engineer

    Diverse Lynx • Bentonville, AR, United States
    Full-time
    Well versed with Hadoop, Spark, Cloud, PythonScala and Java, Streaming, Kafka, Backend, J2EE.You evangelize an extremely high standard of code quality, system reliability, and performance.You have ...Show more
    Last updated: 30+ days ago • Promoted
    Sr. Software Engineer CSMS / OCPP SME (Bentonville, AR)

    Sr. Software Engineer CSMS / OCPP SME (Bentonville, AR)

    Cedent • Bentonville, AR, United States
    Full-time
    Must sit in or relocation to Bentonville (hybrid schema) by day 1.Level (10+ years of experience) with Java, Python or Golang framework. EV vehicle and Charger station working Protocol & Various Gui...Show more
    Last updated: 30+ days ago • Promoted