Senior Machine Learning Engineer

Air; Oregon; Hybrid; USD 170,000 - 240,000; Full time

Job summary

Air is seeking a Senior Machine Learning Engineer to build the infrastructure for language models and AI systems, enabling rapid experimentation from training to deployment and observability. You will own end-to-end ML lifecycle, including datasets, evaluation, model serving, and production reliability.

This engineering-heavy role blends ML research with infrastructure, supporting teams to move models from experimentation to production at scale.

Qualifications

• 5+ years building production ML systems and infrastructure.
• Experience with LLM fine-tuning and post-training workflows.
• Strong Python programming for production-grade software.
• Experience with AWS, GCP, Azure and GPU-based workloads.

Responsibilities

• Design and build LLMOps infrastructure for production models.
• Build scalable training and fine-tuning infra for LLMs.
• Develop pipelines for supervised fine-tuning and parameter-efficient methods.
• Create data pipelines for training, evaluation, and synthetic data generation.
• Implement model registries, artifact management, and versioning.
• Develop experiment management infrastructure for model comparisons.
• Build automated evaluation pipelines for readiness for production.
• Design observability for training and inference, including metrics and logs.
• Optimize training/inference for GPU utilization and cost.
• Support product/AI engineering integration with multiple models/providers.

Skills

Production ML systems

Python programming

Kubernetes & containers

Cloud platforms

ML pipeline tooling

Model versioning & experiments

LLM fine-tuning workflows

Distributed systems

Observability

Tools

Docker

MLflow

Kubernetes

Terraform

Git

Job description

Company Description

Air is the leader in Enterprise Readiness. Our mission is to establish readiness as a real-time condition that is continuously achieved. Today, a dangerous Readiness Gap exists between what the front line needs and what is delivered. Our AI-native platform, Air Enterprise Readiness, aligns development, production, delivery, and sustainment into one coordinated execution system for government agencies and industrial suppliers. By revealing true capacity, exposing real constraints, coordinating resources, and executing at the speed of operational demands, the front line gets what it needs to succeed.

Job Description

We are seeking an experienced Senior Machine Learning Engineer to join our AI/ML team and build the infrastructure that powers the development, evaluation, deployment, and continuous improvement of our language models and AI systems. As our AI capabilities expand, we need robust infrastructure for moving models from experimentation into production. This role will own critical parts of that lifecycle, including LLMOps, fine-tuning infrastructure, model evaluation, dataset pipelines, experiment management, model serving, and production observability. In order to do this job well: This is an engineering-heavy ML role. You will build platforms and infrastructure that allow AI engineers and researchers to rapidly experiment with models, datasets, and training techniques while maintaining the reproducibility, scalability, and reliability required for production systems. You will work across the full model lifecycle - from dataset creation and experimentation through training, evaluation, deployment, monitoring, and iteration. This role is a full-time position based in our Pittsburgh, PA office or open to Remote Opportunities. This role may require up to 25% travel, including periodic travel to our Pittsburgh, PA and Arlington, VA offices for team collaboration, planning activities, and in-person meetings.

Scope of Responsibilities

• Design and build LLMOps infrastructure supporting the development, evaluation, deployment, and continuous improvement of production language models.
• Build scalable training and fine-tuning infrastructure for commercial and open-weight language models.
• Develop pipelines supporting supervised fine-tuning, parameter-efficient fine-tuning, preference optimization, and other post-training techniques.
• Build infrastructure for distributed training and GPU-accelerated ML workloads.
• Develop data pipelines for training, fine-tuning, evaluation, and synthetic data generation.
• Build systems for dataset versioning, lineage, quality validation, transformation, and reproducible experimentation.
• Develop experiment management infrastructure that enables engineers to compare models, datasets, hyperparameters, prompts, and training techniques.
• Build automated evaluation pipelines that determine whether new models or model versions are ready for production deployment.
• Design model registries, artifact management, versioning, and promotion workflows across development and production environments.
• Build and operate scalable model-serving and inference infrastructure for open-weight and fine-tuned models.
• Develop abstractions that allow product and AI engineering teams to use multiple models and inference providers without tightly coupling applications to a single model or vendor.
• Build observability for model training and inference, including metrics, tracing, logging, resource utilization, model quality, latency, throughput, and cost.
• Optimize training and inference workloads for GPU utilization, throughput, latency, reliability, and infrastructure cost.
• Build automated workflows for model deployment, rollback, canarying, and production validation.
• Investigate model and infrastructure failures across data pipelines, training jobs, inference services, distributed systems, and production environments.
• Evaluate emerging models, training techniques, inference frameworks, and ML infrastructure and determine where they can improve our production systems.
• Partner closely with AI engineers building agentic systems to provide the model, evaluation, and training infrastructure required to continuously improve those systems.

Qualifications

* U.S. Citizenship is required

Required Skills

• 5+ years of experience building production machine learning systems, ML infrastructure, distributed systems, or similar technical systems.
• Deep experience designing, building, and operating production ML infrastructure or ML platforms.
• Experience building infrastructure for training, fine-tuning, evaluating, deploying, and monitoring large language models or other large-scale deep learning models.
• Experience with LLM fine-tuning and post-training workflows, including techniques such as supervised fine-tuning, LoRA/QLoRA or other parameter-efficient approaches, and preference optimization.
• Strong understanding of the modern LLM lifecycle, including data preparation, training, evaluation, model artifacts, deployment, inference, monitoring, and iteration.
• Experience building reproducible ML pipelines involving dataset versioning, experiment tracking, model versioning, and automated evaluation.
• Experience building and operating production GPU infrastructure across AWS, GCP, Azure, or dedicated GPU providers, including training and/or inference workloads.
• Strong understanding of distributed systems and the challenges involved in running computationally intensive ML workloads at scale.
• Strong programming experience in Python and experience building production-quality software.
• Deep experience with containers, Kubernetes, and cloud platforms such as AWS, GCP, or Azure.
• Experience designing scalable APIs, services, asynchronous workloads, and data-processing pipelines.
• Strong understanding of observability and operational reliability for production ML systems.
• Comfortable debugging failures across training code, datasets, models, GPUs, distributed systems, and cloud infrastructure.
• Able to move between ML experimentation and infrastructure engineering, understanding the needs of researchers and AI engineers while building systems that make those workflows scalable and reproducible.
• Comfortable working in a rapidly evolving field where tooling, models, and best practices change quickly.

Desired Skills

• Current possession of a U.S. security clearance, or the ability to obtain one with our sponsorship
• Experience in or exposure to the nuances of a startup or other entrepreneurial environment
• Experience building secure code execution environments or sandboxes for AI agents.
• Experience with multi-agent architectures, agent-to-agent communication, or distributed agent execution.
• Experience with fine-tuning, post-training, reinforcement learning, or synthetic data generation.
• Experience building AI observability, tracing, and debugging infrastructure.
• Experience optimizing inference latency, throughput, GPU utilization, or model-serving costs.
• Experience with AI security, adversarial testing, or securing agentic systems.
• Experience working in government, defense, or other mission-critical environments.

We firmly believe that past performance is the best indicator of future performance. If you thrive while building solutions to complex problems, are a self-starter, and are passionate about making an impact in global security, we're eager to hear from you.Air is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, national origin, sexual orientation, gender identity, disability and protected veterans status or any other characteristic protected by law.