Software Engineer - Axion Data Engine and ML Ops

Applied Intuition Applied Intuition · Robotics · Sunnyvale, CA · Government

Software Engineer role focused on building the data engine for training perception models, including edge applications and cloud data ingestion pipelines. The role involves optimizing ML model execution on edge and cloud, developing MLOps tooling, and integrating foundation models for data automation.

What you'd actually do

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

Skills

Required

  • 5+ years of relevant work experience
  • modern ML infrastructure
  • data-centric AI approaches
  • running large-scale jobs on GPUs
  • microservices
  • databases for data-oriented software
  • U.S. citizenship
  • eligibility to obtain a security clearance

Nice to have

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

What the JD emphasized

  • U.S. citizenship (legally required)
  • eligibility to obtain a security clearance

Other signals

  • training perception models
  • MLOps
  • edge application
  • cloud data engine
  • continual improvement of models