Skip to content

Sr Machine Learning Engineer, AI Research

185k – 215kUnited StatesRemote4+ YOE
Summary

Design, train, and productionize ML models and pipelines for AI-enabled security/observability platforms. Requires 4+ years experience with Python, PyTorch/TensorFlow, and modern ML techniques.

About the role

Responsibilities

  • Design, train, and evaluate machine learning models across research and applied AI initiatives
  • Run rapid, iterative experiments to test hypotheses and surface insights that drive model improvements
  • Collaborate closely with researchers and engineers to translate cutting-edge academic advances into practical, production-ready systems
  • Build and maintain robust ML pipelines for data ingestion, feature engineering, model training, and evaluation
  • Optimize model performance through fine-tuning, hyperparameter search, and architecture experimentation
  • Contribute to a culture of rigorous experimentation; track results, document findings, and share learnings with the broader team
  • Stay current with the latest developments in ML and AI research, and proactively identify opportunities to apply them

Requirements

  • Bachelor's degree in Computer Science, Mathematics, Statistics, or a related field with 4+ years of industry or research experience (Master's or PhD a plus)
  • Deep hands-on experience training and evaluating ML models, including language models
  • Strong proficiency in Python and ML frameworks such as PyTorch or TensorFlow
  • Familiarity with MLOps tooling and infrastructure (e.g., MLflow, Weights & Biases, Kubeflow, or similar)
  • Solid understanding of modern NLP, computer vision, and/or reinforcement learning techniques
  • Strong ability to move fast without sacrificing rigor; know when to prototype and when to productionize
  • Excellent communication skills with the ability to clearly present experimental results to both technical and non-technical stakeholders

Nice-to-Haves

  • Master's or PhD degree
Skills
PythonPyTorchTensorFlowMLflowWeights & BiasesKubeflowNLPComputer VisionReinforcement LearningMLOps
Similar roles at this salary range
All ML Engineering jobs →
Watershed

Staff Software Engineer, AI Platform

Technical leader building agent infrastructure, observability, evals, and guardrails for production AI systems at Watershed. Requires 6+ years backend/platform/AI engineering experience and production TypeScript systems.

202k – 255kSan Francisco, CAML EngineeringOn-site6+ YOELLMsTypeScript
Nuro

Software Engineer, ML Infrastructure

Build and scale ML infrastructure platform for autonomous vehicle development, focusing on automated resource provisioning, high-performance workload scheduling, and petabyte-scale data processing pipelines.

160k – 241kMountain View, CAML EngineeringOn-site3+ YOERaySlurm
Nuro

Senior Software Engineer, ML Infrastructure

Build and scale ML infrastructure platform for autonomous vehicle model development, focusing on automated resource provisioning, high-performance workload scheduling, and petabyte-scale data processing pipelines.

194k – 291kMountain View, CAML EngineeringOn-site4+ YOERayAWS
Nuro

Software Engineer, ML Infrastructure, Optimization

Build and optimize ML infrastructure for autonomous vehicles, focusing on model optimization, compilers, and deployment across the autonomy stack. Requires 2+ years in ML optimization and strong Python/C++/CUDA skills.

160k – 241kMountain View, CAML EngineeringOn-site2+ YOEC++JAX
Nuro

Senior/Staff Software Engineer, ML Infrastructure, Optimization

Build and optimize ML infrastructure for autonomous vehicles, focusing on model optimization, compilers, and deployment of large models to Nuro's self-driving fleet. Requires 3+ years ML optimization experience and strong Python/C++/CUDA skills.

194k – 352kMountain View, CAML EngineeringOn-site3+ YOEC++JAX