Skip to content
ArmisArmisUnited States

Principal Machine Learning Engineer

Build and optimize AI/ML pipelines for data processing, model training, and deployment. Collaborate with data scientists to implement scalable AI solutions on cloud infrastructure.

184k – 250k
Remote7+ YOEML Engineering

About the role

Responsibilities

  • Design, build, and maintain robust AI pipelines for data processing, model training, and deployment.
  • Collaborate with data scientists and engineers to implement cutting-edge AI solutions.
  • Optimize existing AI infrastructure for performance, scalability, and reliability.
  • Troubleshoot and resolve issues in the AI pipeline environment.
  • Document and maintain best practices for AI pipeline development and deployment.
  • Develop and implement Minimum Viable Product (MVP) code for AI initiatives.
  • Collaborate with other engineers to transition MVP code to production-ready systems.

Requirements

  • Bachelor's or Master's degree in Computer Science, Engineering, or a related field, or equivalent professional experience.
  • Proven experience in building and managing AI/ML pipelines.
  • Proficiency in Python and other relevant programming languages.
  • Familiarity with cloud platforms (AWS, Azure, GCP) and containerization technologies (Docker, Kubernetes).
  • Strong understanding of data processing and machine learning concepts.

Compensation

  • Salary range: $184,000 - $250,000
  • Bonus and Pre-IPO Equity

Skills

PythonAWSAzureGCPDockerKubernetesMachine LearningData ProcessingAi PipelinesMLOps

Similar roles

ML Engineering jobs
PointClickCare

Principal AI Platform Engineer (US)

PointClickCareUnited States

Build and maintain core GenAI infrastructure including model gateways, vector databases, observability tools, and integration with agent workflows. Requires extensive AI platform experience, Kubernetes, OpenTelemetry/MLFlow, and preferred LLM serving frameworks.

179k – 199k
Remote7+ 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
On-site7+ 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
Hybrid5+ 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
Remote15+ YOEML Engineering
Upstart

Principal Machine Learning Engineer

UpstartUnited States

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
Remote7+ YOEML Engineering