Machine Learning Engineer (PyTorch)

Responsibilities

• Complete up to 3 assigned technical tasks involving low-level PyTorch work.
• Ensure code runs correctly through full runtime (including unattended/extended 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 (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.

Why Apply

$150 per task — complete up to 3 tasks
Fully remote, work on your own schedule
Low time commitment: ~2 hours active work per task
Rolling onboarding starting next week.