About the Role Mercor is partnering with a leading AI research lab to support a Frontier Code Agents project. This role focuses on evaluating and improving frontier AI coding models through realistic machine learning engineering workflows and structured technical assessments.…
PyTorch Internals Expert — Project Torch
Job description
AfterQuery is seeking experienced machine learning, systems, and distributed systems engineers with deep, internals-level PyTorch expertise to complete advanced technical tasks for Project Torch.
This fully remote contract opportunity focuses on low-level PyTorch development, including custom autograd implementations, CUDA and tensor extensions, ATen internals, and distributed training systems. Selected contributors will complete up to three technical assignments that evaluate real-world expertise in PyTorch internals.
Key Responsibilities
Complete Advanced PyTorch Tasks
Work independently on up to three technical assignments involving low-level PyTorch development and runtime optimization.
Develop and Validate Solutions
Ensure all code executes successfully through both active development and extended unattended runtime.
Work with PyTorch Internals
Tasks may involve areas such as:
- Custom autograd functions
- CUDA and tensor extensions
- ATen-level development
- Distributed training internals
- Runtime optimization
- Low-level PyTorch implementation
Ideal Qualifications
- Professional or research experience working full-time with PyTorch.
- Demonstrated expertise with PyTorch internals.
- Ability to work independently on advanced technical assignments.
- Access to GPU-enabled hardware or a cloud GPU environment.
Preferred Qualifications
- Background in machine learning, systems engineering, or distributed systems.
- Experience with distributed training internals.
- Compiler or graph-level optimization experience.
- Numerical or runtime optimization expertise.
- Contributions to PyTorch or related open-source projects.
Compensation
- $65 per completed task (up to 3 tasks)
- $100 bonus for successfully completing all three tasks and passing review
Project Details
- Fully remote
- Contract engagement
- Rolling onboarding
- Most assignments require approximately 2 hours of active work, with additional unattended runtime
About AfterQuery
AfterQuery is an AI research laboratory exploring the frontiers of artificial intelligence through novel datasets and experimentation.
Backed by Y Combinator and Box Group, AfterQuery partners with leading AI laboratories to advance next-generation AI systems.
You will be redirected to the company's website to complete your application.