A
PyTorch Internals Engineer (Task-Based, Remote)
Remote
AI Summary
This is a task-based role for PyTorch experts focusing on low-level internals. Complete up to 3 short PyTorch tasks for $150 each, requiring deep custom autograd, CUDA/tensor extensions, or ATen-level experience. Fully remote with rolling onboarding starting next week, offering flexibility and a low time commitment.
Key Highlights
$150 per task with up to 3 tasks available
Fully remote work with flexible scheduling
Requires deep, internals-level PyTorch experience
Key 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
Technical Skills Required
Benefits & Perks
$150 per task
Fully remote
Low time commitment
Nice to Have
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
Job Description
AfterQuery is sourcing ML, systems, and distributed systems engineers with deep, internals-level PyTorch experience — custom autograd, CUDA/tensor extensions, ATen-level work, and distributed training internals. This is a task-based role: complete up to 3 short PyTorch tasks, $150 per task, fully remote, ~2 hours active work per task, rolling onboarding starting next week.
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
- 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
Interested in remote work opportunities in Machine Learning & AI? Discover Machine Learning & AI Remote Jobs featuring exclusive positions from top companies that offer flexible work arrangements.
- 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)
Browse our curated collection of remote jobs across all categories and industries, featuring positions from top companies worldwide.
- 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
AfterQuery is a research lab investigating the boundaries of artificial intelligence through novel datasets and experimentation. We're backed by top investors, including Y Combinator and Box Group, and support all leading AI labs.
Similar Jobs
Explore other opportunities that match your interests
Visa Sponsorship
Relocation
Remote
Job Type
Full-time
Experience Level
Mid-Senior level
Understanding Recruitment
United Kingdom
Visa Sponsorship
Relocation
Remote
Job Type
Contract
Experience Level
Mid-Senior level
MBN Solutions
United Kingdom
Visa Sponsorship
Relocation
Remote
Job Type
Full-time
Experience Level
Not Applicable
Jobgether
United State