AI/ML Research Engineer

Peraton Labs, Inc.; Red Bank (NJ); On-site; USD 146,000 - 234,000; Full time

Job summary

Peraton Labs is seeking an AI Research Engineer to lead the design and implementation of the machine learning core for generative-model research and development. The role focuses on model formulation, training/inference pipelines, and containerized deployment.

The ideal candidate has a research background in causal inference, probabilistic programming, or Bayesian methods and excels in a software engineering discipline.

Qualifications

• MS or PhD in CS/EE/Statistics/Applied Math or related field is required.
• 2+ years of applied ML research experience in causal inference, probabilistic programming, Bayesian methods, or learned optimization.
• Strong Python software engineering skills with production-quality PyTorch usage on GPUs.
• Experience turning research prototypes into maintainable, containerized code that runs unattended.
• Knowledge of multi-objective and constrained optimization applied to design spaces.

Responsibilities

• Design, implement, and train causal graph models and planning algorithms.
• Build ML infrastructure: training/inference pipelines, experiment tracking, dataset loaders.
• Package trained models for inference as containerized components with defined API.
• Contribute to synthetic data generation and simulation campaigns for training coverage.
• Present results at reviews and coordinate with academic subcontractors.

Skills

Python

PyTorch

ML research

Causal inference

Probabilistic programming

Bayesian methods

Multi-objective optimization

Experiment management

Reading research papers

US Citizenship

Education

MS or PhD in CS/EE/Statistics/Applied Math

Tools

GPU profiling

Container tooling

Job description

About Peraton

Peraton is a next-generation national security company that drives missions of consequence spanning the globe and extending to the farthest reaches of the galaxy. As the world's leading mission capability integrator and transformative enterprise IT provider, we deliver trusted, highly differentiated solutions and technologies to protect our nation and allies. Peraton operates at the critical nexus between traditional and nontraditional threats across all domains: land, sea, space, air, and cyberspace. The company serves as a valued partner to essential government agencies and supports every branch of the U.S. armed forces. Each day, our employees solve the most daunting challenges that our customers face. Visit peraton.com to learn how we're keeping people around the world safe and secure.

About Peraton

Peraton is a next-generation national security company that drives missions of consequence spanning the globe and extending to the farthest reaches of the galaxy. As the world's leading mission capability integrator and transformative enterprise IT provider, we deliver trusted, highly differentiated solutions and technologies to protect our nation and allies. Peraton operates at the critical nexus between traditional and nontraditional threats across all domains: land, sea, space, air, and cyberspace. The company serves as a valued partner to essential government agencies and supports every branch of the U.S. armed forces. Each day, our employees solve the most daunting challenges that our customers face. Visit peraton.com to learn how we're keeping people around the world safe and secure.

About The Role

Peraton Labs is seeking an AI Research Engineer to lead the design and implementation of the machine learning core for extensive research and development into generative models. The role focuses on the research-to-code path for model generation, planning, and optimization, including formulating models, building the training and inference pipelines, integrating with synthetic and laboratory datasets, and delivering trained, containerized frameworks.

The ideal candidate combines a research background in causal inference, probabilistic programming, or Bayesian methods with strong software engineering discipline. They must be comfortable working in a domain where model outputs must be explainable and verifiable. Familiarity with communications and digital signal processing is preferred but not required at the outset; the ability to learn the domain quickly and collaborate closely with RF, DSP, and formal methods engineers is required.

Key Responsibilities

• Design, implement, and train causal graph models, planning algorithms, as well as architect approaches to multi-objective optimization
• Build and maintain the ML infrastructure: training and inference pipelines, experiment tracking, model versioning, dataset loaders for synthetic and real data, and the objective and interface
• Package trained models for inference as containerized components that conform to the specified API, working with the Lead System Integrator to deliver regular code drops and major code revisions
• Contribute to the design of synthetic data generation and simulation campaigns so that training data covers the objective space the models must generalize across
• Present research results at program design reviews, site visits, and PI workshops; author technical sections of monthly status reports and design documentation; coordinate with academic subcontractor researchers on shared model components

Qualifications

Required Qualifications

• 5+ years of experience and an MS or PhD in Computer Science, Electrical Engineering, Statistics, Applied Mathematics, or a related technical field
• 2+ years of applied machine learning research experience with demonstrated work in at least one of: causal inference and structural causal models, Bayesian networks and probabilistic graphical models, probabilistic programming (Pyro, NumPyro, Stan, PyMC, or similar), or learned optimization and planning
• Strong Python software engineering skills with production-quality use of PyTorch, including custom model implementation, training loop design, and performance profiling on GPUs
• Experience translating research prototypes into maintainable, tested, containerized code that runs unattended in an environment the developer does not control
• Some working knowledge of multi-objective and constrained optimization (evolutionary, Bayesian, gradient-based, or combinatorial methods) and experience applying it to structured design spaces
• Experience with experiment management, reproducible ML pipelines, and dataset versioning on a multi-person research team
• Ability to read and implement methods from the current research literature, and to communicate model behavior and limitations clearly to non-ML engineers and government stakeholders
• US Citizenship

Desired Qualifications

• Prior work on IARPA, DARPA, or similar government research programs with independent T&E, leaderboard-based evaluation, and a regular delivery cadence
• Background in communications, digital signal processing, or information theory, including familiarity with modulation, coding, channel models
• Experience with neuro-symbolic methods, program synthesis, or graph generation models (graph neural networks, autoregressive graph generators, or diffusion over graphs)
• Familiarity with symbolic regression (PySR or similar) or with generating executable code or domain-specific language artifacts from learned models
• Experience with hardware-constrained inference (model compression, quantization, or deployment on embedded or edge compute)
• Publications in causal inference, probabilistic machine learning, or machine learning for wireless communications
• Willingness and ability to obtain Secret security clearance

Details

Target Salary Range: $146,000 - $234,000. This represents the typical salary range for this position. Salary is determined by various factors, including but not limited to, the scope and responsibilities of the position, the individual's experience, education, knowledge, skills, and competencies, as well as geographic location and business and contract considerations. Depending on the position, employees may be eligible for overtime, shift differential, and a discretionary bonus in addition to base pay.

Benefits Statement: Peraton offers eligible employees a variety of benefits including medical, dental, vision, life, health savings account, short/long term disability, EAP, parental leave, 401(k), paid time off (PTO) for vacation, and company paid holidays. A full listing of available benefits can be viewed at https://www.careers.peraton.com/benefits.

EEO

Equal opportunity employer, including disability and protected veterans, or other characteristics protected by law.