Note: The job is a remote job and is open to candidates in USA. AfterQuery Experts is a research lab investigating the boundaries of artificial intelligence through novel datasets and experimentation. They are seeking a Machine Learning Engineer with deep PyTorch experience to complete technical tasks involving low-level PyTorch work.

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

Skills

• 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)
• 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

Benefits

• $100 bonus for completing all three tasks and passing review
• Fully remote

Company Overview