[Remote] Machine Learning Engineer (PyTorch)
Note: The job is a remote job and is open to candidates in USA. reputed company 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 reputed company a rolling reputed company 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 reputed company 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-reputed company libraries
Benefits
- $100 bonus for completing reputed company three tasks and passing review
- Fully remote
Company Overview