Machine Learning Engineer (PyTorch)
AfterQuery Experts
full-remotemidcontractbackenddata Greater Boston 65 days ago via LinkedIn
Commission-based
See how well this job matches your profile
Sign up to get an AI match score and generate a tailored application in seconds.
Get your match scoreTags
PyTorchCustom AutogradCUDAATenTensor ExtensionsDistributed TrainingGPUSystems EngineeringOpen Source
About the role
Role Overview
Task-based Machine Learning Engineer (PyTorch) role requiring deep, internals-level PyTorch expertise. Complete up to 3 short PyTorch tasks (up to $150 per task), with fully remote work and rolling onboarding starting next week.
Responsibilities
- Complete up to 3 assigned technical tasks focused on low-level PyTorch work
- Ensure submitted code runs correctly through the full runtime, including extended/unattended execution
- Work independently within a rolling onboarding schedule
Requirements
- Full-time professional or research experience with PyTorch
- Proven internals-level PyTorch experience, such as:
- Custom autograd
- Tensor/CUDA extensions
- ATen-level work
- Access to suitable GPU-enabled hardware (local GPU machine or cloud instance)
Preferred Qualifications
- Background in ML, systems engineering, or distributed systems engineering
- Experience with distributed training internals
- Experience with compiler/graph-level work or numerical/algorithmic runtime optimization
- Contributions to PyTorch or adjacent open-source libraries
About AfterQuery Experts
AfterQuery is a research lab exploring the boundaries of artificial intelligence through novel datasets and experimentation. Backed by top investors, including Y Combinator and Box Group, the lab supports and collaborates with leading AI research teams.
Scraped 7/23/2026