Machine Learning Engineer (Pytorch) | $160/hr Remote

At Crossing Hurdles, we work as a referral partner. We refer candidates to Mercor that collaborates with the world’s leading AI research labs to build and train cutting-edge AI models.

Organization: Mercor

Position: PyTorch Operator - ML Engineer

Type: Hourly Contract

Compensation: $100–$160/hour

Location: Remote

Commitment: 10-40hrs/week, flexible, asynchronous

Role Responsibilities (Training support will be provided)

• Design and implement new PyTorch tensor operators in C++/ATen.
• Develop and validate Python bindings ensuring correct gradient propagation and test coverage.
• Create gold standard reference implementations in eager mode for correctness assessment.
• Collaborate asynchronously with CUDA engineers for kernel optimization integration.
• Profile, benchmark, and report performance at operator and computational graph levels.
• Document APIs, assumptions, and performance features for reproducibility.

Ideal Qualifications

• Deep expertise in PyTorch internals, including TensorIterator, dispatcher, and autograd engine.
• Strong skills in modern C++ (C++17+) and template metaprogramming within PyTorch ecosystems.
• Experience creating or extending custom PyTorch ops or backend implementations.
• Familiarity with performance profiling tools and GPU-CPU interplay.
• Excellent written communication and ability to deliver well-documented, modular code.
• Contributions to PyTorch or related open-source projects are highly valued.

Application Process:

1. Upload resume
2. AI interview based on your resume (15 min)
3. Submit form