Skip to content
UpstartUpstartUnited States

Principal Machine Learning Engineer

Leads engineering initiatives to build scalable ML platform infrastructure, including unified embeddings, feature pipelines, and continuous learning systems. Requires 7+ years in applied ML, Python/ML frameworks expertise, and Master's/PhD in quantitative field.

221k – 300k/yr
Remote7+ YOEML Engineering

About the role

Responsibilities

  • Scale ML innovation by building tools, infrastructure, and workflows that dramatically improve the speed and reliability of model development.
  • Work backward from modeling needs to design systems that directly unlock gains in accuracy, efficiency, and scientific productivity.
  • Explore new algorithms and methodologies for our machine learning models and develop tooling to support them.
  • Improve the entire ML lifecycle—from data readiness and feature development through training, evaluation, serving, and monitoring.
  • Automate and standardize operational workflows, enabling scientists to focus on high-leverage modeling and analysis rather than manual pipelines.
  • Define the roadmap for our next generation ML Platform, balancing near-term impact with long-term architectural scalability.
  • Collaborate cross-functionally with Data Engineering, ML Platform, Pricing, and other teams to build reliable, end-to-end ML systems.

Minimum Qualifications

  • 7+ years of hands-on experience in applied machine learning, with strong exposure to production-scale modeling efforts.
  • Demonstrated expertise in end-to-end model development: data prep, feature engineering, training, evaluation, and deployment.
  • Experience working in high-scale, ML-driven product environments—especially in fintech, pricing, or risk modeling.
  • Proficiency in Python and core ML frameworks (e.g., PyTorch, TensorFlow, Scikit-learn, XGBoost).
  • Ability to work autonomously and lead technical direction in ambiguous, high-impact domains.
  • Experience collaborating with cross-functional teams including ML scientists, engineers, and product partners.
  • Ability to bridge engineering and science teams, and influence technical strategy across disciplines.
  • Numerically-savvy and smart with ability to operate at a fast pace.
  • Master’s degree or PhD in a quantitative discipline, or equivalent additional professional experience.

Preferred Qualifications

  • Practical experience optimizing ML workflows using CUDA/GPU acceleration.
  • Background in feature store design, embedding architecture, or synthetic data generation for model training.
  • Proven track record of improving model accuracy in production environments with measurable business outcomes.
  • Familiarity with modern experimentation frameworks, hyperparameter tuning tools, and automated model selection techniques.

Skills

PythonPyTorchTensorFlowscikit-learnXgboostMachine LearningFeature EngineeringKubernetesCUDAFeature Store

Similar roles

ML Engineering jobs
Everlaw

Staff/Principal AI Engineer

EverlawOakland, CA

Leads design and execution of AI strategy for legal discovery platform, guiding engineering teams on LLMs, NLP, and frontier models at massive scale. Requires 6+ years ML/AI experience and 10+ years software engineering.

228k – 340k/yr
Hybrid6+ YOEML Engineering
Shippo

Principal Machine Learning Engineer, ML Platform

ShippoUnited States

Principal ML Platform Engineer builds production ML infrastructure for shipping logistics ML (prediction, fraud, optimization). Requires 15+ years engineering, 4+ years ML production systems, Kubernetes/MLflow expertise.

212k – 287k/yr
Remote15+ YOEML Engineering
Formation Bio

Principal Data Scientist, Medical Imaging

Formation BioNew York, NY +1

Lead development of deep learning models for medical image analysis (MRI, X-ray) to accelerate clinical trials, create AI biomarkers, and integrate imaging AI into drug development workflows. Requires PhD and 5+ years building production-grade medical imaging AI.

205k – 267k/yr
Hybrid5+ YOEML Engineering
Astera

Principal Software Engineer, Neuroscience Platforms

AsteraEmeryville, CA

Lead development of computational infrastructure for large-scale neuroscience research, building real-time stimulus/acquisition systems, high-throughput data pipelines, and cloud tools. Requires exceptional full-stack engineering with real-time systems experience; neuroscience background not required.

200k – 350k/yr
On-site7+ YOEML Engineering
Pinterest

Principal Engineer, AI Platform

PinterestSan Francisco, CA

Principal Engineer setting technical vision and building AI/ML infrastructure for Generative AI and Recommender Systems at Pinterest, scaling to hundreds of millions of inferences per second. Requires deep expertise in distributed systems and proven cross-org technical leadership.

243k – 500k/yr
Hybrid7+ YOEML Engineering