Machine Learning Engineer (PyTorch)
AfterQuery Experts · San Francisco, CA · 1 wk ago
RemoteRemoteAnalyst$150/hrContract
Responsibilities
- Complete up to 3 assigned technical tasks involving low-level PyTorch work.
- Ensure code runs correctly through full runtime, including any extended, unattended runtime.
- Complete tasks independently within a rolling onboarding schedule.
Required Qualifications
- Full-time professional or research experience with PyTorch.
- Demonstrated experience with internals-level PyTorch work — e.g., custom autograd functions, tensor/CUDA extensions, or ATen-level work.
- Access to suitable hardware (GPU-enabled 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.
Pay
$150 per task, up to 3 tasks.
Schedule
Fully remote, rolling onboarding next week.