Senior Machine Learning Engineer, Firefly Foundry

Adobe Adobe · Enterprise · San Jose, CA +2

ML infra engineer to build and optimize infrastructures for large foundation model training on thousands of GPUs, including data processing, PyTorch training, and GPU optimizations with custom CUDA kernels. Focus on generative AI for text, images, and videos.

What you'd actually do

  1. You'll build and optimize infrastructures that power large foundation model training on thousands of GPUs
  2. You will profile GPU utilization, trace inference and training runs and help craft strategies for optimizing our ML model latency.
  3. We'll work together to architect and optimize end-to-end ML pipelines, ensuring they're scalable, efficient, and robust.
  4. You'll dive deep into data to recommend the right models, evaluation metrics, and governance approaches.
  5. Throughout the product lifecycle, you'll engage in architecture, design, deployment, and optimizations of ML models and systems.

Skills

Required

  • 5+ years ML Engineering experience
  • generative AI like LLMs
  • Strong Python
  • deep learning engineering skills
  • training and inferencing with PyTorch or TensorFlow

Nice to have

  • distillation
  • transformers
  • diffusion models
  • generative image and video
  • deployment technologies such as Docker
  • ML Ops
  • ML services
  • cloud platforms like Azure and AWS

What the JD emphasized

  • large scale foundation models
  • PyTorch training infrastructures
  • GPU optimizations
  • custom CUDA kernels

Other signals

  • large scale foundation models
  • PyTorch training infrastructures
  • GPU optimizations
  • custom CUDA kernels