Staff Software Engineer, AI Foundation Model

Merlin; Boston (MA); On-site; USD 180,000 - 260,000; Full time

Benefits offered by this job

Catered lunches

Snacks and beverages

Health, dental, life insurance

Unlimited vacation

401k with match

Job summary

Merlin Labs in Boston, MA, is seeking a senior AI/ML infrastructure engineer to own the Foundation models, training and inference infrastructure, and end-to-end data pipelines for autonomous flight systems.

You will mentor engineers, drive architecture decisions, and collaborate with AI, simulation, and flight software teams to ensure safe, scalable AI deployment in real-world environments.

Qualifications

• 8+ years of software engineering with AI/ML infrastructure
• Experience building production-grade AI foundations and data pipelines
• Proficient in PyTorch/TensorFlow and at least one systems language

Responsibilities

• Strategize Foundation model selection, refactor and maintain core AI models
• Define benchmarks and safety/reliability standards for evaluation
• Architect data pipelines ingesting, labeling and versioning flight data
• Collaborate with AI infra, Simulation and Flight Software teams
• Mentor junior/mid-level engineers and lead design reviews
• Lead scoping for large AI foundation projects with actionable plans
• Evaluate open-source tooling and contribute improvements
• Document architecture decisions and operational runbooks
• Contribute to hiring through interviews and setting engineering bar

Skills

AI/ML infrastructure

Foundation model deployment

Data pipelines & ML data engineering

Distributed training

Pytorch/TensorFlow

Systems language (C++, Rust, Go)

Mentoring engineers

System design & architecture

Tools

Weights & Biases

MLFlow

JAX

TensorRT

Job description

About Merlin

Merlin (NASDAQ: MRLN) is a publicly traded aerospace and defense company building a non-human pilot to deliver full-stack autonomy for any aircraft from takeoff to touchdown. The Merlin Pilot autonomy system powers a growing range of aircraft and mission profiles and has been proven through hundreds of autonomous flights from Merlin's global flight test facilities, including Kerikeri, New Zealand; Quonset Point, Rhode Island; and soon, Bedford, Massachusetts. Headquartered in Boston, Merlin is expanding its organization to accelerate the development and deployment of its autonomy platform, helping customers solve some of aviation's most pressing challenges, from pilot shortages to improving flight safety. Backed by some of the world's leading investors prior to its public listing, Merlin continues to advance the certification and commercialization of autonomous flight across commercial and defense aviation.

About you

You are a senior individual contributor with deep expertise in building the systems that make large-scale machine learning possible in production. You've experience in the selection of optimum foundation models, adopt/tune for target application needs, designed required data pipelines that handle real-world messiness, built training infrastructure that scales reliably, and created evaluation frameworks that give engineers genuine confidence in their development. You care about the craft of software engineering as much as you care about AI models — you know that great AI systems are only as good as the infrastructure underneath them. You enjoy operating with significant autonomy, influencing technical decisions across multiple teams, and mentoring engineers who are earlier in their careers. Most importantly, you want to work on a problem that genuinely matters: putting safe autonomous aircraft into the skies.

Responsibilities

• Technically strategize on Foundation model selection, refactor, design, build, and maintain Merlin's core AI models,training and inference infrastructure, including distributed training pipelines, experiment tracking, and model registry systems.
• Define and drive standards and benchmarks for model evaluation, benchmarking, and regression testing to ensure AI systems meet safety and performance thresholds before deployment.
• Architect and own the foundational data pipeline that ingests, processes, labels, and versions flight data for use across all autonomy and AI teams.
• Identify and resolve systemic technical bottlenecks that slow down AI development velocity across Merlin's engineering organization.
• Collaborate with AI infra, Simulation, and Flight Software teams to define interfaces and shared abstractions that make the broader stack more coherent and maintainable.
• Mentor, build and technically guide junior and mid-level engineers on the AI Foundation team, conducting design reviews and raising the overall quality of the team's output.
• Lead technical scoping and estimation for large AI foundational model projects, breaking ambiguous requirements into actionable engineering plans.
• Evaluate and adopt relevant open-source tooling, frameworks, and research, contributing back where appropriate.
• Document architecture decisions, system designs, and operational runbooks to a standard that supports safety review and long-term maintainability.
• Contribute to hiring by conducting technical interviews and helping define the engineering bar for the AI Foundation team.

Qualifications

• 8+ years of software engineering experience, with a substantial portion focused on state of the art and next-gen AI model development, ML infrastructure, MLOps, data engineering, or AI platform development.
• Experience in development of the AI-first Autonomy software tech stack that includes Perception, Behavior Planning, Prediction and Actuation ( based on Transformer architectures)
• Demonstrated ability to design and deliver large-scale, production-grade systems independently — from initial architecture through deployment and operation.
• Deep proficiency in Pytorch, TensorFlow and at least one systems language such as C++,C, Rust, or Go.
• Hands-on experience with distributed training frameworks (Python extensions, JAX, or equivalent) and the infrastructure required to run them reliably at scale.
• Experience in onboarding trained models into edge environments consisting of High performance compute System-on-Chips.
• Strong background in data pipeline design, including streaming and batch processing, data versioning, and handling high-volume, heterogeneous sensor data.
• Experience building model evaluation and validation frameworks leveraging MLOps frameworks (like Weights and Biases. MLFlow etc) that go beyond accuracy metrics to assess real-world reliability and safety-relevant behavior.
• Proven track record of influencing technical direction across teams and driving alignment on shared infrastructure and standards.
• Excellent written communication skills, with the ability to produce clear design documents and architecture proposals that hold up to rigorous review.
• Comfort working in a fast-moving startup where the scope of problems and priorities evolve as the product matures.

Nice to Have

• Experience in aerospace/automotive autonomy, robotics, or other safety-critical real-time domains where software quality and reliability standards are exceptionally high.
• Development and deployment of AI models on heterogeneous HPCs like NVIDIA Thor/Orin.
• Background in simulation infrastructure, synthetic data generation, or domain randomization for training perception and control models using end to end ecosystems like NVIDIA Cosmos etc., .
• Experience with onboard inference optimization — model quantization, TensorRT, hardware-aware compilation, or deployment on embedded accelerators.
• Contributions to widely used open-source ML or data engineering projects.
• Familiarity with flight data formats, avionics data buses (ARINC 429, MIL-STD-1553), or sensor modalities common in aviation (Camera, LiDAR, radar, IMU).
• Prior staff or principal engineer experience at a high-growth startup or research-driven

This position is based on-site at Merlin HQ in Boston, MA.

Once you’re here, you’ll enjoy a variety of on-site perks designed to make your workday enjoyable and convenient. These include catered lunches featuring a rotating menu of delicious options, an assortment of snacks to keep you fueled throughout the day, and a selection of beverages, including coffee, tea, and other drinks, to keep you refreshed.

Our goal is to create an environment where you can thrive both professionally and personally

Merlin Labs offers an innovative, entrepreneurial, and team-focused startup environment. We also offer a top-tier benefits package (health, dental, life, unlimited vacation, and 401k with match) and work/life integration. Being part of the Merlin team allows you to become part of a small team that supports professional development while working together to achieve our mission.

Merlin Labs is an equal opportunity employer and values diversity. We do not discriminate on the basis of race, religion, color, national origin, genetic information, sex (including pregnancy), gender, gender identity and expression, sexual orientation, age, marital status, military service or obligation or disability status, or any other characteristic protected by law. All job offers are contingent upon the candidate passing background and reference checks.

At this time, we are unable to provide visa sponsorship or consider candidates who require visa transfers. Applicants must be authorized to work in the United States without the need for visa sponsorship now or in the future.

In compliance with federal law, all persons hired will be required to verify identity and eligibility to work in the United States and to complete the required employment eligibility verification form upon hire.

If you require reasonable accommodation in completing an application, interviewing, completing any pre-employment testing, or otherwise participating in the employee selection process, please direct your inquiries to: [e-mail]

Merlin Labs does not accept unsolicited resumes from any source other than directly from candidates.

The compensation range provided is a good‑faith estimate based on a wide range of factors, inclusive of base salary only. The actual offer will be determined by a variety of factors including the candidate's qualifications, skills, experience, education, and training. Highly competitive equity grants are considered part of Merlin's total compensation package. Additionally, Merlin offers top-tier benefits for full-time employees.