Skip to content
Applied Intuition

Software Engineer - Axion Data Engine and ML Ops

125k – 222kSunnyvale, CAOnsite5+ YOE
Summary

Builds and optimizes edge and cloud data pipelines for ML perception models in autonomy, integrates foundation models for labeling automation, and evolves MLOps tooling. Requires 5+ years experience with ML infra, GPUs, microservices.

About the role

Responsibilities

  • Construct optimized data pipelines to run ML models
  • Evolve our data engine architecture to scale high-fidelity labels, reduce annotation costs, and accelerate ML iteration cycles
  • Integrate foundation models (LLMs, VLMs, and multimodal models) to automate and enhance labeling, quality assurance, and data discovery
  • Leverage software-in-the-loop and hardware-in-the-loop testing
  • Interact with the DoD customer to understand their use cases, requirements, and triage needs during field events to deliver a superior customer experience

Requirements

  • 5+ years of relevant work experience
  • Familiarity with modern ML infrastructure, data-centric AI approaches and running large-scale jobs on GPUs
  • Created or worked on microservices and/or databases for data-oriented software
  • A hunger to learn and grow into a position of ownership and impact on a new product team
  • U.S. citizenship (legally required) and eligibility to obtain a security clearance

Nice to Have

  • Full-stack experience React, TypeScript, Python, Golang or similar
  • Experience with Docker, Kubernetes, Opensearch and Postgres
  • Direct experience with foundation models, including LLMs and VLMs, for data automation tasks
  • Background in autonomous driving or robotics perception
  • Experience with active learning, auto-labeling, or human-in-the-loop ML systems

Compensation

Base salary range: $125,000 - $222,000 USD annually, plus equity and benefits.

Skills
ML infrastructureGPUsmicroservicesdatabasesDockerKubernetesPostgresOpensearchLLMsVLMsPythonGolangReactTypeScriptactive learning
Similar roles at this salary range
All ML Engineering jobs →
Chime

AI/ML Engineer

Build and productionize ML models for risk detection and decisioning systems. Requires 1-2 years applied ML experience and familiarity with AWS, model evaluation, and experimentation.

125k – 173kSan Francisco, CAML EngineeringHybridAWSPython
PrizePicks

Machine Learning Platform Engineer

Build and operate the ML platform to productionize models, enable real-time inference, and manage the full ML lifecycle with MLOps best practices. Requires 3+ years platform engineering and 1+ years owning ML systems end-to-end.

135k – 160kUnited StatesML EngineeringRemoteGoKafka
Ai2

Senior Research Engineer

Research Engineer training and scaling flagship open multimodal and agentic models (Olmo, Molmo). Owns end-to-end ML infrastructure, model development, and open-source releases.

147k – 220kSeattle, WAML EngineeringOn-siteJAXMoE
Clickhouse

AI Product Engineer

Build agentic capabilities on a petabyte-scale observability platform. Own the full agent stack including context engineering, tool design, evals, and production reliability for incident investigation.

130k – 230kUnited StatesML EngineeringRemoteMCPSQL
Kodex

ML Engineer

Build and deploy production ML models and pipelines to detect suspicious activity, improve verification accuracy, and support threat intelligence workflows.

150k – 180kUnited StatesML EngineeringRemoteAWSClustering
Apply