Talent.com
PyTorch Operator - ML Engineer
PyTorch Operator - ML EngineerMercor • Ann Arbor, Michigan, US
PyTorch Operator - ML Engineer

PyTorch Operator - ML Engineer

Mercor • Ann Arbor, Michigan, US
17 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 • Ann Arbor, Michigan, US

    Related jobs
    Applied ML Engineer, II - GPU Optimization

    Applied ML Engineer, II - GPU Optimization

    Torc Robotics • Ann Arbor, Michigan, United States
    Full-time
    At Torc, we have always believed that autonomous vehicle technology will transform how we travel, move freight, and do business. A leader in autonomous driving since 2007, Torc has spent over a deca...Show more
    Last updated: 30+ days ago • Promoted
    Machine Learning Signal Processing Engineer

    Machine Learning Signal Processing Engineer

    Leidos Inc • Ann Arbor, MI, United States
    Full-time
    Do you want to join a high performing team that values integrity, innovation, and collaboration with a company whose mission is to make the world safer, healthier, and more efficient through inform...Show more
    Last updated: 30+ days ago • Promoted
    Lansing, Michigan Field Property Claim Specialist

    Lansing, Michigan Field Property Claim Specialist

    The Auto Club Group • Wixom, MI, United States
    Full-time
    It's a great time to join AAA The Auto Club Group!.JOIN THE TEAM COMMITTED TO DRIVING YOUR CAREER FORWARD.Eligible candidates for this role should reside within a commutable distance of Lansing, Mi...Show more
    Last updated: 14 days ago • Promoted
    CUDA Kernel Optimizer - ML Engineer

    CUDA Kernel Optimizer - ML Engineer

    Mercor • Ann Arbor, Michigan, US
    Remote
    Full-time
    Role Overview • • Mercor is engaging advanced CUDA experts who specialize in GPU kernel optimization, performance profiling, and numerical efficiency. These professionals possess a deep mental model o...Show more
    Last updated: 17 hours ago • Promoted • New!
    Applied ML Engineer, I - GPU Optimization

    Applied ML Engineer, I - GPU Optimization

    Torc Robotics • Ann Arbor, Michigan, United States
    Full-time
    At Torc, we have always believed that autonomous vehicle technology will transform how we travel, move freight, and do business. A leader in autonomous driving since 2007, Torc has spent over a deca...Show more
    Last updated: 30+ days ago • Promoted
    CNC Programmer and Machinist

    CNC Programmer and Machinist

    Atlas Copco • New Hudson, MI, US
    Full-time
    Passionate people create exceptional things.At Atlas Copco, we believe in challenging the status quo, always looking for a better way. Our leading-edge technology in self-pierce riveting and industr...Show more
    Last updated: 2 hours ago • Promoted • New!
    Estimator

    Estimator

    Central Conveyor Company, LLC (a member of the Tsubaki Group) • Wixom, MI, US
    Full-time
    Central Conveyor Company (a member of the Tsubaki Group) is a turn-key Systems Integrator with a complete portfolio of conveyance solutions and products for the Automotive, Parcel & Logistics, ...Show more
    Last updated: 30+ days ago • Promoted
    Engineer I Labs - HDS (Willow Run location)

    Engineer I Labs - HDS (Willow Run location)

    NSF International • Ypsilanti, MI, United States
    Full-time
    As an Engineer I for Engineering / Plastics, you will be responsible for conducting a wide variety of testing on plastics, pipes, fittings and other related plumbing products to specific standards an...Show more
    Last updated: 30+ days ago • Promoted
    DevOps Engineer - Onsite Wixom, MI

    DevOps Engineer - Onsite Wixom, MI

    J&b Medical Supply Co • Wixom, Michigan, United States
    Full-time
    A DevOps Engineer plays a critical role in bridging the gap between software development and IT operations.They are responsible for designing and managing automated systems for software deployment,...Show more
    Last updated: 30+ days ago • Promoted
    ML AI Engineer

    ML AI Engineer

    Mice Groups • Ann Arbor, MI, US
    Full-time
    ML / AI Engineer / Contract / Hybrid, 3 days per week onsite in Ann Arbor, MI or Palo Alto, CA / 1 year, extendable.Summary : Simplify vehicle software development and increase developer agility by ...Show more
    Last updated: 30+ days ago • Promoted
    Programmer / Analyst

    Programmer / Analyst

    American Mathematical Society • Ann Arbor, Michigan, United States
    Full-time
    At the American Mathematical Society (AMS) we value excellence, inclusivity and teamwork.We work at the forefront of advancing and connecting the diverse mathematical community worldwide, and we ar...Show more
    Last updated: 30+ days ago • Promoted
    Business Banking Rel Mgr

    Business Banking Rel Mgr

    Old National Bank • Brighton, MI, United States
    Full-time
    US-MI-Ann Arbor | US-MI-Brighton.Lending / Commercial / Consumer / Credit.Old National Bank has been serving clients and communities since 1834. With over $70 billion in total assets, we are a regional po...Show more
    Last updated: 9 days ago • Promoted
    CNA / 3rd Shift

    CNA / 3rd Shift

    BrightSpring Health Services • Holly, MI, United States
    Full-time
    Rehab Without Walls Neuro Rehabilitation.Rehab Without Walls Neuro Rehabilitation.Join Our Team as a Certified Nursing Assistant with Rehab Without Walls!. A transitional rehabilitation program desi...Show more
    Last updated: 10 days ago • Promoted
    Fast Pay - Earn at least $1730 in your first 140 trips, guaranteed.

    Fast Pay - Earn at least $1730 in your first 140 trips, guaranteed.

    Uber • Britton, Michigan, United States
    Full-time +1
    Earn at least $1730 driving with Uber when you complete your first 140 trips in 30 days.Why Uber? Driving is an easy way to boost your income while maintaining the flexibility your schedule require...Show more
    Last updated: 13 days ago • Promoted
    Remote AI Writing Evaluator

    Remote AI Writing Evaluator

    Outlier • Wixom, MI, United States
    Remote
    Full-time
    Earn up to $15 / hour + performance bonuses.Outlier, a platform owned and operated by Scale AI, is looking for.If you're passionate about improving models and excited by the future of AI, this is you...Show more
    Last updated: 1 day ago • Promoted
    Oral and Maxillofacial Surgeon

    Oral and Maxillofacial Surgeon

    TruForm Surgery • Brighton, US
    Full-time
    Join our rapidly growing oral and maxillofacial surgery practice in Southeast / Central Michigan! TruForm Surgery was founded by Army veterans, Drs. Adam and Brittany Eidson with a vision of helping p...Show more
    Last updated: 30+ days ago • Promoted
    Physician

    Physician

    Correctional Medicine Associates • Milan, US
    Full-time
    At CoreCivic, we do more than manage inmates, we care for people.CoreCivic is currently seeking .The Physician provides professional medical direction, oversight and direct patient care in a correc...Show more
    Last updated: 30+ days ago • Promoted
    Civil Project Engineer

    Civil Project Engineer

    Jobot • Brighton, MI, US
    Full-time
    This Jobot Job is hosted by : Brian Perkins.Are you a fit? Easy Apply now by clicking the "Apply Now" button and sending us your resume. Salary : $60,000 - $120,000 per year.Since our inception, we ha...Show more
    Last updated: 30+ days ago • Promoted