PyTorch Operator - ML EngineerMercor • Santa Ana, California, US
PyTorch Operator - ML Engineer
Mercor • Santa Ana, California, US
22 hours 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 • Santa Ana, California, US
Related jobs
S State College, Fullerton, CA, 92831, US.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability to lawfully obtain access to such...Show more S Placienta Ave, Fullerton, CA, 92831, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's abilit...Show more 
The Electro Optical InfraredSystems (EOIS) line of business within DRS has locations in Dallas and Austin, TX, Melbourne, FL, and Cypress, CA.
EOIS develops, manufactures, and supports infrared and ...Show more
The Tactical Recon and Strike (TRS) division at Anduril develops aerial small drones (Group 1-3) and all equipment to test, deploy and operate.
The team is responsible for taking products like Ghost...Show more
Anduril Industries is a defense technology company with a mission to transform U.By bringing the expertise, technology, and business model of the 21st century's most innovative companies to the def...Show more City of Industry, CA, 91744, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability to lawfu...Show more 
CNC Lathe Machine Operator (2nd Shift).We are seeking a reliable and detail-oriented Machine Operator (2nd shift) to set up, operate, and maintain machinery in a safe and efficient manner.The ideal...Show more S State College, Fullerton, CA, 92831, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's abilit...Show more 
Crank It Up : We're Hiring Machine Pros Like You.Every shop needs someone to run the machines - and that's where you come in.
As a Granulation Machine Operator, you'll keep production rolling while c...Show more The Electro Optical InfraredSystems (EOIS) line of business within DRS has locations in Dallas and Austin, TX, Melbourne, FL, and Cypress, CA.
EOIS develops, manufactures, and supports infrared and ...Show more The Tactical Recon and Strike (TRS) team at Anduril develops aerial small drones (Group 1-3) and all equipment to test, deploy and operate.
The team is responsible for taking products like Ghost, An...Show more The Tactical Recon and Strike (TRS) team at Anduril develops aerial small drones (Group 1-3) and all equipment to test, deploy and operate.
The team is responsible for taking products like Ghost, An...Show more 
Anduril Industries is a defense technology company with a mission to transform U.By bringing the expertise, technology, and business model of the 21st centurys most innovative companies to the defe...Show more
Individuals assigned to the Machine Shop shall be able to use blueprints, drawings, sketches, or computer-aided design (CAD) and manufacturing files to program by using computer software such as Ma...Show more S State College, Fullerton, CA, 92831, US.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's ability to lawfully obtain access to such...Show more S Placienta Ave, Fullerton, CA, 92831, US.Remote Work Schedule Availability?.This position entails access to export-controlled items and employment offers are conditioned upon an applicant's abilit...Show more Been searching for "machine operator jobs near me " that offer more than just a paycheck? Gear up and join our team cause we are.
Safety-minded and steady on the team.Competitive Pay : Weekly depos...Show more 
You will be a part of a small, production-minded ML team based in Orange County / Oakland.You’ll collaborate with other engineers and researchers to develop, evaluate, and help deploy vision models f...Show more
CNC Operator IV - 2nd Shift
Howmet • Fullerton, CA, United States
Full-time
Last updated: 14 days ago • Promoted
Tooling Specialist V, Machine Build- 1st Shift
Howmet • Fullerton, CA, United States
Full-time
Last updated: 30+ days ago • Promoted
Principal Mechanical Engineer (Thermal Analyst)
Leonardo DRS • Cypress, CA, United States
Full-time
Last updated: 30+ days ago • Promoted
Senior Machine Learning Data Engineer
Anduril Industries • Costa Mesa, California, United States
Full-time
Last updated: 30+ days ago • Promoted
Senior Software Engineer, Machine Learning (C++ or Python)
Anduril Industries • Costa Mesa, CA, United States
Full-time
Last updated: 24 days ago • Promoted
Tooling Specialist II, Tool Prep
Howmet • City of Industry, CA, United States
Full-time
Last updated: 30+ days ago • Promoted
CNC Lathe Machine Operator (2nd Shift)
Element Materials Technology • Huntington Beach, CA, United States
Full-time
Last updated: 30+ days ago • Promoted
Tooling Specialist III- 2nd Shift
Howmet • Fullerton, CA, United States
Full-time
Last updated: 30+ days ago • Promoted
Granulation Machine Operator
Staffmark Group • Anaheim, CA, United States
Full-time
Last updated: 10 days ago • Promoted
Algorithms Software Engineer
Leonardo DRS • Cypress, CA, United States
Full-time
Last updated: 30+ days ago • Promoted
Senior Machine Learning / Computer Vision Engineer (C++ or Python)
Anduril Industries • Costa Mesa, California, United States
Full-time
Last updated: 30+ days ago • Promoted
Senior Machine Learning Engineer (C++ or Python)
Anduril Industries • Costa Mesa, California, United States
Full-time
Last updated: 30+ days ago • Promoted
Senior Software Engineer, Machine Learning (C++ or Python)
Slope • Costa Mesa, CA, United States
Full-time
Last updated: 1 day ago • Promoted
CNC Operator and Programmer - Full Time
Disneyland Resort • Santa Ana, California, US
Full-time
Last updated: 9 days ago • Promoted
Tooling Manufacturing Engineer
Howmet • Fullerton, CA, United States
Full-time
Last updated: 1 day ago • Promoted
CNC Operator I
Howmet • Fullerton, CA, United States
Full-time +1
Last updated: 30+ days ago • Promoted
Blending Operator
Staffmark Group • Anaheim, CA, United States
Full-time
Last updated: 10 days ago • Promoted
Machine Learning Engineer
Kinetic • Santa Ana, California, United States
Full-time
Last updated: 13 days ago • Promoted