Be an early applicantLess than 25 applicants

Company

Original Job Post

Adobe · 6 days ago

Senior ML Engineer, AI/ML Platform & Data

San Francisco, CA

Full-time

Onsite

Senior Level

$171K/yr - $325K/yr

5+ years exp

Wonder how qualified you are to the job?

Maximize your interview chances

ConsultingEnterprise Software

H1B Sponsorship

Actively Hiring

Growth Opportunities

Insider Connection @Adobe

Discover valuable connections within the company who might provide insights and potential referrals, giving your job application an inside edge.

Responsibilities

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 optimize distributed training frameworks leveraging GPUs to improve performance and scalability. Improve resiliency, elasticity, data loading and provide out-of-the-box support for GPU optimization methods such as FP8, FSDP and model parallelism

Write high quality, product level code that is easy to maintain and test following standard methodologies

Collaborate closely with ML Researchers and Machine Learning Engineers to accelerate the training of the cutting-edge ML models

Keep track of the latest innovation in academia and open-source community to implement rapid adoption of pioneering technologies to improve the performance of the ML platform

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

Qualification

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

PythonSystems DevelopmentSDKsInfrastructureModel ServingOrchestrationGPU ResourcesMachine LearningDistributed PytorchProblem-SolvingCritical ThinkingAnalyticalCommunicationRelationship BuildingTeamworkKubeFlowMLFlowRaySageMakerNvidia HPCPytorch distributedMPIMegatronHorovodAI training frameworks

Required

PhD or Master’s in computer science or related field and 5+ years relevant 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

Preferred

Experience with KubeFlow, MLFlow, Ray, SageMaker, or similar

Experience with Nvidia HPC

Experience with Pytorch distributed, MPI, Megatron, Horovod and other AI training frameworks

Benefits

Annual Incentive Plan (AIP)

Company

Adobe

Glassdoor

4.3

Adobe is a software company that provides its users with digital marketing and media solutions.

Founded in 1982

San Jose, California, USA

10,001+ employees

http://www.adobe.com

H1B Sponsorship

Adobe has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Trends of Total Sponsorships

2023 (621)

2022 (795)

2021 (802)

2020 (522)