Staff Software Engineer, AI Foundation Model

You are a senior individual contributor with deep expertise in building the systems that make large-scale machine learning possible in production. Youve 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.

• Technically strategize on Foundation model selection refactor design build and maintain Merlins core AI modelstraining 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 Merlins 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 teams 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.

• 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 CC 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.

• 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