What you will Do - Design, develop, and maintain robust AI/ML infrastructure solutions to support the training and deployment of large-scale AI models, using Kubernetes and Python on AWS cloud
- Implement and improve distributed training frameworks leveraging GPUs to improve performance and scalability. Improve resiliency, elasticity, data loading and provide out-of-the-box support for FSDP and model parallelism
- Help train better models by improving orchestration and scheduling, scaling the number of jobs, faster experimentation with AutoML and similar
- Collaborate with data scientists and ML researchers to streamline the model training pipeline and ensuring efficient resource utilization
- Drive innovation in infrastructure practices to support pioneering machine learning research and development
What you need to succeed - PhD or Master's in computer science or related field and 5+ years of hands-on industry experience
- Proven proficiency with Python and developing systems, frameworks and SDKs
- Experience with infrastructure and understanding of model serving, training, orchestration, and management of GPU resources
- Experience with machine learning and distributed Pytorch
- Strong critical thinking, analytical and quantitative problem-solving ability
- Excellent communication, relationship skills and a strong teammate
While not required, it's an added plus if you also have: - Experience with KubeFlow, MLFlow, Ray, SageMaker, or similar
- Experience with Pytorch distributed, MPI, Megatron, Horovod and other AI training frameworks
#FireflyGenAI
Expected Pay Range:Our compensation reflects the cost of labor across several U.S. geographic markets, and we pay differently based on those defined markets. The U.S. pay range for this position is $172,500 -- $306,625 annually. Pay within this range varies by work location and may also depend on job-related knowledge, skills, and experience. Your recruiter can share more about the specific salary range for the job location during the hiring process.
In California, the pay range for this position is $211,800 - $306,625In Washington, the pay range for this position is $201,000 - $291,150
At Adobe, for sales roles starting salaries are expressed as total target compensation (TTC = base + commission), and short-term incentives are in the form of sales commission plans. Non-sales roles starting salaries are expressed as base salary and short-term incentives are in the form of the Annual Incentive Plan (AIP).
In addition, certain roles may be eligible for long-term incentives in the form of a new hire equity award.
