Machine Learning Engineer (Defense AI)

TL;DR

Machine Learning Engineer (AI/Defense): Designing and building scalable ML services for enrichment workflows with an accent on model training pipelines and high-performance inference APIs. Focus on optimizing models using ONNX, vLLM, and TensorRT to achieve low-latency, high-throughput performance in production.

Location: Remote (Must have U.S. Person status)

Salary: $150,000 - $215,000 + equity

Company

Defense technology company building agentic AI to deter adversaries and simulate campaigns for decision makers.

What you will do

• Design and build scalable ML services for enrichment, including training pipelines and inference APIs.
• Deploy and optimize models using ONNX, vLLM, and TensorRT for low-latency performance.
• Collaborate with product teams to define data requirements and feature engineering strategies.
• Develop monitoring and observability systems to ensure model quality and reliability.
• Research and integrate emerging ML techniques for efficient inference and large-scale processing.

Requirements

• 5+ years of experience deploying ML systems in production.
• Proficiency with PyTorch, TensorFlow, or Jax.
• Expertise in model deployment technologies like Kubernetes and Ray.
• Experience scaling ML services for large data volumes with strict latency requirements.
• Strong software engineering skills in distributed systems, APIs, and cloud infrastructure.
• U.S. Person status is required for access to U.S. only data systems.

Culture & Benefits

• Competitive salary with equity.
• Comprehensive health, dental, and vision insurance.
• Remote-friendly work environment with WeWork access.
• Unlimited PTO and company-wide year-end time off.
• 401(k) match and lifestyle/wellbeing stipends.
• Paid parental leave and child/pet care reimbursement during travel.