Senior ML Research Engineer
Adobe
full-time
Posted on:
Location Type: Hybrid
Location: Seattle • California • Washington • United States
Visit company websiteExplore more
Salary
💰 $172,500 - $306,625 per year
Job Level
About the role
- Write and maintain high-performance GPU kernels using Triton or CUDA
- Support scaling of large-scale training runs
- Implement state-of-the-art inference techniques to reduce latency
- Conduct deep-dive bottleneck analysis to identify stalls
- Collaborate on the 'Model Factory' pipeline for fault-tolerant training jobs
Requirements
- Master’s or PhD's degree in Computer Science, Electrical Engineering, or a related field with a focus on Parallel Computing or Systems
- Proficiency in Python and C++
- Experience working with PyTorch or JAX, especially in distributed setups
- Strong grasp of GPU architecture and its impact on ML workloads
- Experience contributing to open-source efficiency libraries
- Familiarity with containerization and job scheduling in a headless Linux environment
Benefits
- Health insurance
- 401(k) matching
- Flexible working hours
- Paid time off
- Remote work options
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard Skills & Tools
GPU kernelsTritonCUDAPythonC++PyTorchJAXParallel Computingdeep-dive bottleneck analysisopen-source efficiency libraries
Soft Skills
collaborationproblem-solving
Certifications
Master’s degreePhD