Staff Machine Learning Engineer, ML Frameworks

Adobe

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
San Jose, CASeattle, WASan Francisco, CA
Salary
$211,800–$306,625 / yr
Posted
9 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $229k
This role $259k
$170k most similar roles pay here $321k

This role pays more than 80% of similar roles. Most pay $202,412–$254,750 — the shaded band above. At the midpoint, this role pays about $259k versus about $229k for comparable roles.

Based on 240 similar postings.

Employer

About Adobe

Adobe Inc. is a global software company known for creative and multimedia software products including Photoshop, Illustrator, Acrobat, and its cloud-based Creative Cloud and Document Cloud suites. Industry: Creative & Digital Experience Software

Adobe currently has 218 open roles on FindRole.

Listed pay typically runs $187,100–$270,950 across 216 roles with salary data.

Most-posted roles

View all roles at Adobe

At a glance

TL;DR · Staff Machine Learning Engineer, ML Frameworks

Staff Machine Learning Engineer - ML Frameworks will join the team to design, develop, and maintain robust AI/ML infrastructure solutions for training and deploying large-scale models. The role involves implementing distributed training frameworks using GPUs to improve performance, scalability, and resiliency while providing support for FSDP and model parallelism. You will collaborate with data scientists and researchers to streamline training pipelines, optimize resource utilization, and drive innovation in infrastructure practices for machine learning research. Key technical requirements include proficiency in Python, experience with distributed PyTorch, and expertise in managing GPU resources, orchestration, and model serving. The role specifically utilizes Kubernetes and AWS cloud environments. Preferred skills include familiarity with KubeFlow, MLFlow, Ray, SageMaker, MPI, Megatron, or Horovod. This position focuses on the technical infrastructure required to support generative AI models within a creative content ecosystem.

What does a Machine Learning Engineer earn in California?

Median $246394 from 172 postings across 28 companies.

See salary data

What you'll do

  • Design, develop, and maintain robust AI/ML infrastructure solutions using Kubernetes and Python on AWS.
  • Implement and improve distributed training frameworks leveraging GPUs to enhance performance and scalability.
  • Improve system resiliency, elasticity, data loading, and provide support for FSDP and model parallelism.
  • Optimize orchestration and scheduling to increase the number of jobs and accelerate AutoML experimentation.
  • Streamline the model training pipeline in collaboration with data scientists and ML researchers.
  • Ensure efficient resource utilization across all machine learning infrastructure components.
  • Drive innovation in infrastructure practices to support pioneering machine learning research and development.

What we're looking for

  • Must possess a PhD or Master’s degree in computer science or a related field.
  • Requires 5+ years of hands-on industry experience.
  • Must have proven proficiency with Python and developing systems, frameworks, and SDKs.
  • Requires experience with infrastructure, model serving, training, orchestration, and management of GPU resources.
  • Must have experience with machine learning and distributed PyTorch.
  • Requires strong critical thinking, analytical, and quantitative problem-solving abilities.
  • Must possess excellent communication, relationship skills, and be a strong teammate.

More like this

Similar roles

Staff Machine Learning Engineer

Intuit

Mountain View, CA 152 days ago $202,500$274,000
Python TensorFlow Keras Spark SQL NLTK NumPy Pandas AWS GCP Git GitHub Machine Learning Operations Data Pipelines R Data Structures Algorithms
7+ yrs exp

Staff Machine Learning Engineer

Intuit

Mountain View, CA 147 days ago $202,500$274,000
Python NLTK NumPy Pandas TensorFlow Keras R Spark SQL Git GitHub CUDA cuDNN AWS GCP Machine Learning Feature Engineering
6+ yrs exp

Staff Machine Learning Engineer

GEICO

Palo Alto, CA +1 98 days ago $130,000$260,000
Generative AI LLMs Python Java C++ C# Kubernetes CICD Kafka Spark Ray Airflow Temporal PostgreSQL MongoDB Cassandra Snowflake elastic search Qdrant
6+ yrs exp

Staff Machine Learning Engineer

Adobe

San Jose, CA 77 days ago $211,800$306,625
Python PyTorch TensorFlow JAX Hugging Face Ray Spark LLMs multimodal foundation models Vector Search Information Retrieval MLOps Distributed Training A/B Testing Transformer Architectures Representation Learning Metric Learning search, ranking

Staff Machine Learning Engineer

Arm Holdings

Austin, TX 32 days ago $249,900$338,100
Machine Learning Python TensorFlow PyTorch ARM Architecture GPU Neural Networks Computer Architecture hardware aware optimization Deep Learning
Hybrid

Staff Machine Learning Engineer

Intuit

Mountain View, CA 17 days ago $230,000$250,000
AI ML LLM RAG Python PyTorch TensorFlow Pandas NumPy LangChain AWS SageMaker Kubernetes Microservices JavaScript Java NoSQL System Design
8+ yrs exp