Skip to content
Ambient.aiAmbient.aiRedwood City, CA

Senior Software Engineer, AI Infrastructure

Build and optimize scalable AI infrastructure for real-time inference, evaluation, and continuous improvement of LLMs, LVMs, computer vision, and multimodal models on large-scale video data. Requires 4+ years production ML systems experience, strong Python skills, and expertise in inference optimization and model serving.

168k – 205k
Hybrid4+ YOEML Engineering

About the role

What you'll do

  • Design, build, and maintain cutting-edge AI infrastructure for real-time computer vision, LLM, LVM, and multimodal inference workloads.
  • Build scalable systems for running state-of-the-art models across large volumes of video and sensor data.
  • Optimize inference performance across latency, throughput, GPU utilization, reliability, and cost.
  • Develop robust evaluation harnesses and benchmarking systems to measure model quality, system performance, regressions, and production readiness.
  • Build infrastructure for continuous model evaluation, experimentation, and deployment.
  • Partner with research scientists to productionize the latest advances in computer vision, LLMs, LVMs, RAG, and multimodal AI.
  • Improve model-serving architecture, including batching, caching, routing, quantization, model parallelism, and hardware utilization.
  • Develop data engines and feedback loops for collecting training data, evaluating model behavior, and continuously improving AI performance.
  • Create reliable observability, monitoring, and debugging tools for production AI systems.
  • Help define best practices for deploying, evaluating, and operating AI systems in real-world enterprise environments.

What you'll bring

  • 4+ years of industry experience building infrastructure, distributed systems, machine learning platforms, or production AI systems.
  • BS/MS in Computer Science or a related technical field, or equivalent practical experience.
  • Strong programming background, especially in Python, with solid software engineering fundamentals.
  • Experience designing and building scalable machine learning infrastructure for training, inference, evaluation, and deployment.
  • Hands-on experience running deep learning models in production, ideally including LLMs, LVMs, vision-language models, or multimodal models.
  • Strong understanding of inference optimization techniques, including batching, caching, quantization, parallelism, memory optimization, GPU utilization, and latency reduction.
  • Experience with model-serving frameworks or systems such as vLLM, Triton Inference Server or similar technologies.
  • Experience building evaluation frameworks, test harnesses, benchmarks, regression tests, or model-quality measurement systems.
  • Strong background in machine learning and deep learning; computer vision experience is a strong plus.
  • Experience designing data engines or pipelines for collecting, managing, and curating training and evaluation data.
  • Familiarity with integrating advanced AI systems such as LLMs, LVMs, RAG pipelines, embedding models, or multimodal models into production applications.
  • Experience with cloud infrastructure, containers, orchestration, distributed systems, and GPU-based workloads.
  • Strong collaboration and communication skills, with the ability to work effectively with research scientists, product teams, infrastructure teams, and stakeholders.
  • Proactive problem-solving ability, a strong ownership mindset, and adaptability to incorporate new AI technologies and methodologies.

Nice to Have

  • Experience operating large-scale GPU infrastructure or distributed inference systems.
  • Experience with CUDA, NCCL, PyTorch, TensorRT, ONNX, or similar ML systems technologies.
  • Experience with video understanding, real-time computer vision, multimodal AI, or physical-world AI systems.
  • Experience with model compression, speculative decoding, distillation, pruning, or low-latency serving techniques.
  • Experience with prompt evaluation, model regression testing, human-in-the-loop evaluation, or automated quality gates.
  • Familiarity with retrieval-augmented generation, vector databases, embedding models, re-rankers, or search infrastructure.
  • Experience building internal ML platforms or tools used by researchers and applied ML teams.

Skills

PythonLLMsLvmMultimodal AiInference OptimizationvLLMTriton Inference ServerPyTorchTensorRTCUDAGPUDistributed SystemsRAGComputer Vision

Similar roles

ML Engineering jobs
Rippling

Senior Software Engineer, Backend - AI Platform

RipplingSan Francisco, CA +2

As a Senior Software Engineer on the AI Platform team, you will own end-to-end technical design and implementation for high-impact AI platform initiatives. You will build scalable agent infrastructure, design observable and self-correcting systems, and mentor junior engineers.

168k – 280k
Hybrid5+ YOEML Engineering
Drata

Senior Applied Research Engineer

DrataSan Francisco, CA

Senior Applied Research Engineer driving AI system quality through experimentation and evaluation of RAG, retrieval, and reasoning systems. Requires 5+ years applied ML/NLP experience with strong Python and evaluation methodology skills.

167k – 226k
Hybrid5+ YOEML Engineering
Drata

Senior AI Engineer, Agent Harness

DrataSan Francisco, CA

Senior AI Engineer to design, build, and scale agentic AI systems using LLMs for compliance automation. Own end-to-end development of production LLM + retrieval + agent workflows with focus on responsible AI.

167k – 226k
Hybrid5+ YOEML Engineering
Mercury

Senior Machine Learning Operations Engineer

MercurySan Francisco, CA +2

Build and operate Mercury's real-time ML inference platform for fraud risk decisioning. Own model deployment, observability, and lifecycle tooling with strong backend Python fundamentals.

167k – 208k
Hybrid5+ YOEML Engineering
Talkiatry

Senior AI Engineer

TalkiatryUnited States

Build full-stack AI systems including agentic workflows, RAG pipelines, and production infrastructure for mental healthcare applications. Requires 2+ years software engineering experience and 1+ year with LLMs or agentic AI.

170k – 195k
Remote2+ YOEML Engineering