PyTorch Operator - ML EngineerMercor • St. Louis, Missouri, US
No longer accepting applications
PyTorch Operator - ML Engineer
Mercor • St. Louis, Missouri, US
1 day ago
Job type
- Full-time
- Remote
Job description2) 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
###
- 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. ###
Create a job alert for this search
Ml Engineer • St. Louis, Missouri, US
Related jobs
ADDMAN is seeking a CMM Programmer Operator to join the Quality Team in Wright City, MO near St.This is an exciting position that plays a key role in ensuring production parts meet customer require...Show more
Leidos National Security Sector combines technology-enabled services and mission software capabilities in the areas of cyber, logistics, security operations, and decision analytics to support our d...Show more
Were Hiring : Senior Project Lead Engineer for EPIC Systems | Arnold, MO.Duties, Knowledge, and Responsibilities : .Expertise in machine vision hardware, vision concepts and solutions.Provide technica...Show more
Machine Learning Engineer - Model Evaluations, Public Sector.The Public Sector ML team at Scale deploys advanced AI systems-including LLMs, agentic models, and multimodal pipelines-into mission-cri...Show more
Role : Data Integration Engineer.Location : St Louis, MO (hybrid).Data Integration Engineer : Someone who is familiar with both Azure Data Factory and Azure API Management.
Also has experience with DB ...Show more
The Renter Solutions department at one of our clients has an exciting opportunity for an.This team is responsible for the retention and maintenance of driver and loyalty profile information, ensuri...Show more
At Boeing, we innovate and collaborate to make the world a better place.We’re committed to fostering an environment for every teammate that’s welcoming, respectful and inclusive, with great opportu...Show more
Watlow is a global technology and manufacturing leader who provides world class engineering expertise through innovative thermal products and systems, enabling our customers to thrive.We are making...Show more
Machine Learning Engineer - Model Evaluations, Public Sector.Louis, MO; New York, NY; Washington, DC.The Public Sector ML team at Scale deploys advanced AI systems—including LLMs, agentic models, a...Show more
At PwC, our people in data and analytics engineering focus on leveraging advanced technologies and techniques to design and develop robust data solutions for clients.
They play a crucial role in tra...Show more
Job Title : Senior MuleSoft Developer.Seeking a hands-on, client-facing Senior MuleSoft Developer with strong communication skills who can triage and remediate critical integration defects.This role...Show more
Our mission at Sinclair & Rush is to provide innovative solutions for our customers that add value to their products.We offer a safe and environmentally friendly workplace for our employees.We ...Show more
We know that you can’t have great.Because we know our future relies on our deep understanding and relentless focus on our people, we live by our mission : A passion for our people.If you think you c...Show more
Job Title : Senior MuleSoft Developer.Seeking a hands-on, client-facing Senior MuleSoft Developer with strong communication skills who can triage and remediate critical integration defects.This role...Show more
Known for being a great place to work and build a career, KPMG provides audit, tax and advisory services for organizations in today's most important industries.
Our growth is driven by delivering re...Show more
We are seeking a Software Engineer to support our client's intelligence analysis products and capabilities.This position involves providing software development capabilities to enhance existing and...Show more A leading AI technology firm in Washington DC is seeking a Machine Learning Engineer to design automated evaluation pipelines.
The role involves working with advanced AI systems in government settin...Show more 
Leidos National Security Sector combines technology-enabled services and mission software capabilities in the areas of cyber, logistics, security operations, and decision analytics to support our d...Show more
CMM Programmer
ADDMAN Group • St. Louis, MO, United States
Full-time
Last updated: 30+ days ago • Promoted
Mid-Level Software Engineer
Leidos Inc • St. Louis, MO, United States
Full-time
Last updated: 30+ days ago • Promoted
Automation Engineer (Arnold)
Calaway Habeck Talent Consulting • Arnold, MO, US
Full-time +1
Last updated: 2 hours ago • Promoted • New!
Machine Learning Engineer - Model Evaluations, Public Sector
Scale AI, Inc. • St. Louis, MO, United States
Full-time
Last updated: 16 days ago • Promoted
Data Integration Engineer
E-Solutions • St Louis, MO, United States
Full-time
Last updated: 21 days ago • Promoted
Software Engineer 2 - Data Modeling
Apex Systems • St Louis, MO, United States
Full-time
Last updated: 3 days ago • Promoted
Core Processing Lead Senior Engineer
Boeing • Berkeley, Missouri, United States
Permanent +1
Last updated: 3 days ago • Promoted
CO-OP ENG 2026
Watlow • St. Louis, MO, United States
Full-time
Last updated: 30+ days ago • Promoted
Machine Learning Engineer - Model Evaluations, Public Sector
Scale AI • St. Louis, MO, United States
Full-time
Last updated: 14 days ago • Promoted
Forward Deployed Software Engineer-Palantir Foundry-Manager
PwC • St Louis, MO, United States
Full-time
Last updated: 4 days ago • Promoted
Senior MuleSoft Developer (St Louis)
Next Quarter • St Louis, MO, United States
Permanent
Last updated: less than 1 hour ago • Promoted • New!
Machine Operator
Sinclair and Rush Inc. • Arnold, MO, USA
Full-time
Quick Apply
Last updated: 30+ days ago
Sr. AI / ML Engineer II (6121)
Metrostar • Maplewood, Missouri, United States
Full-time
Last updated: 30+ days ago • Promoted
Senior MuleSoft Developer
Next Quarter • St Louis, MO, US
Permanent
Last updated: 13 days ago • Promoted
Integration Support Engineer (GoAnywhere & MuleSoft)
KPMG • St Louis, MO, United States
Full-time
Last updated: 30+ days ago • Promoted
4004 Mid Level Software Engineer
Procession Systems • St Louis, MO, United States
Full-time
Last updated: 3 days ago • Promoted
Public Sector ML Engineer : Robust Evaluation Pipelines
Scale AI • St. Louis, MO, United States
Full-time
Last updated: 14 days ago • Promoted
Mid-Level Software Engineer
Leidos • St Louis, MO, United States
Full-time
Last updated: 4 days ago • Promoted