Staff ML Infrastructure Architect

Adobe Inc.; San Jose (CA); On-site; USD 180,000 - 260,000; Full time

Job summary

Adobe Inc. is seeking a senior AI/ML infrastructure engineer to advance the Firefly training and deployment platform.

You will design robust systems using Kubernetes on AWS, optimize distributed training, and collaborate with data scientists and researchers to accelerate experimentation. You bring a PhD or Masters in CS and 5+ years of industry experience, with strong Python skills, GPU resource management, and distributed PyTorch expertise.

Qualifications

• PhD or Masters in computer science or related field; 5+ years industry experience.
• Proficiency with Python and building systems, frameworks, and SDKs.
• Experience with GPU resource management, training orchestration.
• Experience with ML and distributed PyTorch.
• Strong analytical, problem-solving, and quantitative skills.
• Excellent communication and teamwork.

Responsibilities

• Design, develop, and maintain robust AI/ML infrastructure for training and deployment of large-scale models on AWS using Kubernetes and Python.
• Improve distributed training performance and scalability with GPUs; support FSDP and model parallelism.
• Scale orchestration and scheduling, accelerate AutoML experiments and similar.
• Collaborate with data scientists and ML researchers to optimize the training pipeline and resource utilization.
• Drive infrastructure innovation to support pioneering ML research and development.

Skills

Python

Distributed training

Model serving

Analytical skills

Communication

Education

PhD or Master’s in CS

Tools

Kubernetes

AWS

PyTorch

Job description

Adobe Inc. is seeking a senior AI/ML infrastructure engineer to advance the Firefly training and deployment platform.

You will design robust systems using Kubernetes on AWS, optimize distributed training, and collaborate with data scientists and researchers to accelerate experimentation. You bring a PhD or Masters in CS and 5+ years of industry experience, with strong Python skills, GPU resource management, and distributed PyTorch expertise.