Skip to content
ZooxZooxFoster City, CA

Software Engineer, ML Performance Optimization

Drive ML performance optimization initiatives to make autonomous driving models faster and more efficient using distributed training, quantization, distillation, and profiling tools.

192k – 257k/yr
On-site4+ YOEML Engineering

About the role

Responsibilities

  • Design, implement, and operate cutting-edge ML Training OR Inference performance optimization techniques to scale VLM, VLA, and Foundational models and deploy them efficiently in robotaxis.
  • Collaborate closely with cross-functional teams, including ML researchers, software engineers, data engineers, and hardware engineers, to define requirements and align on architectural decisions.

Requirements

  • 4+ years of total experience, including 2+ years of working on large-scale model training or inference platforms.
  • Experience with training frameworks like PyTorch, leveraging GPUs efficiently for distributed model training.
  • Experience with GPU-accelerated inference using TensorRT or similar frameworks.
  • Experience using profiling tools like NVIDIA's Nsight or PyTorch's Profiler for identifying model training and serving bottlenecks.
  • Proficient in Python or C++.

Nice-to-Haves

  • Experience with distributed training techniques, quantization, distillation, and pruning.
  • Work with SOTA accelerators and inference optimization frameworks.

Skills

PyTorchTensorRTPythonC++Nvidia NsightPytorch ProfilerDistributed TrainingQuantizationDistillationPruning

Similar roles

ML Engineering jobs
Zoox

AI Engineer

ZooxFoster City, CA

Build and deploy production GenAI/LLM applications and intelligent agents to automate workflows across Procurement, Supply Chain, Legal, Finance, HR, and Marketing. Requires 5+ years engineering experience including 2+ years production LLMs, Python, LangChain/LlamaIndex, and cloud AI/vector DB tools.

190k – 250k/yr
Hybrid5+ YOEML Engineering
Norm Ai

Software Engineer, Applied AI

Norm AiNew York, NY

Build and ship reliable, scalable LLM-powered AI agents for high-stakes legal workflows at an agentic law platform. Requires 4+ years software/AI engineering experience, strong Python skills, production ML/LLM integration experience, and AI fluency.

190k – 280k/yr
Hybrid4+ YOEML Engineering
Sesame

ML Engineer

SesameSan Francisco, CA +2

Research Engineer building and deploying production voice and multimodal ML models. Requires expert PyTorch, large-scale model training experience, and shipping user-facing ML systems.

190k – 320k/yr
On-site5+ YOEML Engineering
Parafin

Data Scientist, Underwriting

ParafinSan Francisco, CA

Develops and deploys production machine learning models for underwriting small business loans, focusing on risk assessment, pricing optimization, and automated systems. Requires strong statistical reasoning, ML expertise, engineering skills, and business acumen.

190k – 300k/yr
HybridML Engineering
Eve

AI Engineer

EveSan Mateo, CA

Build and deploy AI systems for legal workflows, fine-tuning models on domain-specific data, advancing reasoning capabilities, and integrating with product features. Requires Bachelor's/Master's in ML/AI/CS and proven production ML experience.

195k – 300k/yr
HybridML Engineering