Remote Jobs RockRemote Jobs Rock

Staff Engineer, World Model Development

📅 Sep 17
Deep LearningPytorchWorld ModelsSequence Prediction
Apply Now

📜 Description

  • Design, train, and evaluate world models predicting aircraft state, environment, and traffic under candidate actions.
  • Build rollout tools that let planners evaluate candidate plans against predicted outcomes.
  • Provide calibrated uncertainty and out-of-distribution signals for safety monitoring.
  • Define where predictions are reliable across flight regimes, weather, traffic densities, and configurations.
  • Train models in simulation, validate against flight data, and work with the Data/Sim team to improve simulator fidelity.
  • Benchmark learned dynamics against physics-based and flight-controls models, and maintain reproducible evaluations.

🛠️ Requirements

  • Degree in Computer Science, Artificial Intelligence, Data Science, Computer Engineering, Applied Math, or a related subject.
  • 10+ years building and training deep learning models, with meaningful work on sequence, dynamics, video or trajectory prediction.
  • Strong PyTorch and a solid grasp of the training stack — distributed training, mixed precision, experiment tracking, debugging a run that has silently gone wrong.
  • Working knowledge of dynamical systems, state estimation or control; you can read a flight dynamics model and know what your network is and is not replacing.
  • Demonstrated rigor in evaluation and uncertainty quantification.
  • Comfort with messy real-world sensor and telemetry data.
  • Model-based RL, latent dynamics models or learned simulators.
  • Experience with trajectory prediction in autonomous driving or robotics.
  • Aerospace background: flight dynamics, aircraft performance or air traffic.
  • Multimodal fusion across vision, state and structured mission context.

✨ Benefits

  • Health benefits
  • Dental benefits
  • Life insurance
  • Flexible paid time off
  • 401(k) with match
  • Work/life integration
Full job description
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; Smithfield, 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.
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; Smithfield, 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. Merlin Labs offers an innovative, entrepreneurial, and team-focused startup environment. We also offer a top-notch benefits package (health, dental, life, flexible paid time off, 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: people@merlinlabs.com   Merlin Labs does not accept unsolicited resumes from any source other than directly from candidates.

About You:

  • You want to build a model that understands what happens next. Not a classifier, not a chatbot — a learned model of an aircraft moving through an environment, good enough that a planner can imagine several futures and pick one. You are as interested in where your model is wrong as in where it is right, because in this application the second question is the one that matters.
  • You'll own the learned predictive models of the aircraft, its environment and other actors in it — and the rollout machinery that turns those models into evaluated candidate futures.
  • Responsibilities:

  • Design, train and evaluate world models that predict the evolution of aircraft state, environment and other traffic under candidate action sequences.
  • Build the rollout and imagination machinery that lets a planner evaluate candidate plans against predicted outcomes before committing to one.
  • Own uncertainty: calibrated predictive uncertainty and out-of-distribution signals are required outputs of your models, consumed directly by the safety monitoring layer.
  • Characterize model validity boundaries rigorously — the flight regimes, weather, traffic densities and configurations where predictions can be trusted, and where they degrade.
  • Work the sim-to-real gap in both directions with the Data/Sim team: train in simulation, validate against flight data, and feed discrepancies back into simulator fidelity.
  • Benchmark learned dynamics against Merlin's existing physics-based models and flight-controls models; be honest about where the classical approach wins.
  • Contribute to the shared eval harness and hold the line on reproducibility: every result traceable to a data version, a config and a seed.
  • Qualifications:

  • Degree in Computer Science, Artificial Intelligence, Data Science, Computer Engineering, Applied Math, or a related subject.
  • 10+ years building and training deep learning models, with meaningful work on sequence, dynamics, video or trajectory prediction.
  • Strong PyTorch and a solid grasp of the training stack — distributed training, mixed precision, experiment tracking, debugging a run that has silently gone wrong.
  • Working knowledge of dynamical systems, state estimation or control; you can read a flight dynamics model and know what your network is and is not replacing.
  • Demonstrated rigor in evaluation and uncertainty quantification.
  • Comfort with messy real-world sensor and telemetry data.
  • Nice to Have:

  • Model-based RL, latent dynamics models or learned simulators.
  • Experience with trajectory prediction in autonomous driving or robotics.
  • Aerospace background: flight dynamics, aircraft performance or air traffic.
  • Multimodal fusion across vision, state and structured mission context.
  • Logistics:

  • We welcome remote applicants for this role, with a preference for candidates based in or willing to relocate to Boston, MA for a hybrid work schedule at our HQ.
  • Chime

    Staff AI/ML Engineer

    🕒 8 days ago
    Chime👥 10,000+ employees🏢 Financial Services

    Lead end-to-end development of production machine learning systems for cash-flow underwriting, shaping technical direction and turning transaction data into fair, accurate credit decisions and measurable member outcomes.

    Machine LearningDeep LearningCredit UnderwritingCash-flow Modeling
    Wayve

    Principal Machine Learning Engineer GAIA

    🕒 17 days ago
    Wayve👥 501 - 1000 employees🏢 Computer Software

    Lead post-training and closed-loop development for Gaia, Wayve’s driving world model, improving long-horizon generation, model reliability, and downstream simulation impact through experimentation, data curation, and technical direction.

    Machine LearningLarge-scale Model TrainingModel Post-trainingFine-tuning
    Wayve

    Machine Learning Engineer, Application Software

    🕒 20 days ago
    Wayve👥 501 - 1000 employees🏢 Computer Software

    Lead the development and production deployment of machine-learning driving behaviours, improving autonomous driving performance through model architecture, data pipelines, evaluation, and cross-team technical leadership.

    Machine LearningDeep LearningSequential ModelsControl
    ServiceNow

    Staff Machine Learning Engineer - VoIP Infrastructure

    🕒 8 days ago
    ServiceNow👥 10,000+ employees🏢 Computer Software

    Designs and operates VoIP infrastructure for AI-driven voice workloads, integrating telephony platforms with LLM features while improving reliability, observability, deployment practices, and production support.

    VoipSIP/RTPKamailioRtpengine
    ServiceNow

    Staff Machine Learning Engineer - VoIP Infrastructure

    🕒 8 days ago
    ServiceNow👥 10,000+ employees🏢 Computer Software

    Design and operate VoIP infrastructure for AI-driven voice workloads, integrating telecom platforms with LLM features while improving reliability, observability, and production support across distributed systems.

    VoipSIP/RTPKamailioRtpengine

    Trusted by Remote Workers