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
Employment
Full-time
Posted
39 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

How this pay compares to similar roles

Similar $228k
$173k $289k
below market most similar roles pay here above market

This listing doesn't post a salary. Most similar roles pay $202,187–$254,750.

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 282 open roles on FindRole.

Listed pay typically runs $148,100–$282,100 across 76 roles with salary data.

Most-posted roles

View all roles at Adobe

At a glance

TL;DR · Staff Machine Learning Engineer, ML Frameworks

The Staff Machine Learning Engineer - ML Frameworks joins the team responsible for Firefly, a family of generative AI models. In this role, you will design, develop, and maintain robust AI/ML infrastructure solutions to support the training and deployment of large-scale models. Your daily responsibilities include implementing distributed training frameworks to improve performance and scalability, enhancing resiliency, and optimizing data loading for FSDP and model parallelism. You will collaborate with researchers to streamline training pipelines and improve resource utilization. The position requires proficiency in Python, distributed Pytorch, and managing GPU resources on AWS using Kubernetes. You will solve technical problems related to orchestration, scheduling, and scaling jobs for pioneering machine learning research and development within a commercially safe generative AI product area.

What does a Machine Learning Engineer earn in California?

Median $235750 from 200 postings across 34 companies.

See salary data

What you'll do

  • Design, develop, and maintain robust AI/ML infrastructure solutions using Kubernetes, Python, and AWS.
  • Implement and improve distributed training frameworks to enhance GPU performance and scalability.
  • Provide out-of-the-box support for FSDP and model parallelism.
  • Improve orchestration and scheduling to scale the number of jobs and accelerate AutoML experimentation.
  • Streamline the model training pipeline to ensure efficient resource utilization.
  • Drive innovation in infrastructure practices to support pioneering machine learning research and development.

What we're looking for

  • PhD or Master’s in computer science or a related field.
  • 5+ years of hands-on industry experience.
  • Proven proficiency with Python and developing systems, frameworks, and SDKs.
  • Experience with infrastructure, 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.
  • Experience with KubeFlow, MLFlow, Ray, SageMaker, or similar (preferred).
  • Experience with Pytorch distributed, MPI, Megatron, Horovod, and other AI training frameworks (preferred).

More like this

Similar roles

Staff Machine Learning Engineer

Capital One Financial

Plano, TX +1 47 days ago
Machine Learning Python C++ Scala C Dask RAPIDS NumPy Pandas Scikit-learn Distributed Computing Data Pipelines High Performance Computing
10+ yrs exp

Staff Machine Learning Engineer

Arm Holdings

Austin, TX 71 days ago $198,100–$268,000
Machine Learning Python TensorFlow PyTorch ARM Architecture Computer Architecture Deep Learning GPU Neural Networks
3+ yrs exp Hybrid

Staff Machine Learning Engineer

Intuit

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

Staff Machine Learning Engineer

Adobe

San Jose, CA +1 34 days ago
Generative AI Agentic AI Computer Vision Multimodal Learning LLMs Transformers Diffusion Models Vision-Language Models (VLMs) Python PyTorch AWS Azure Docker Kubernetes Model Context Protocol (MCP) APIs
5+ yrs exp

Staff Machine Learning Engineer

Capital One Financial

Plano, TX +1 28 days ago $269,100–$307,200
Python Java GoLang C++ PyTorch TensorFlow Spark Ray Kubernetes AWS GCP Azure Dask RAPIDS Pandas NumPy Scikit-learn Transformers RNNs CNNs LSTMs Data Pipelines

Staff Machine Learning Engineer

Coinbase

Remote 12 days ago $218,025–$256,500
Python LLMs Deep Learning Graph Neural Networks NLP Machine Learning Information Retrieval Data Mining Statistics Gradient-Boosted Trees Regression Data Workflows Production ML Systems Observability Responsible AI
8+ yrs exp Remote