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

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 and 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: (Takes 20 min) • Upload resume • AI interview based on your resume (15 min) • Submit form