Senior Machine Learning Engineer | AI/ML Maestro

Infinity Labs • Huntsville, AL

3d • Onsite

About The Position

This position is open for a Senior AI/ML Engineer specializing in reinforcement learning (RL) and defense-related modeling and simulation. Candidates must meet the general requirements below to be considered. Additional qualifications for senior-level applicants are listed throughout the description. You wield artificial intelligence like a six-shooter, deploying machine learning algorithms with the precision of a marksman. You’re a problem solver who doesn’t flinch at complex military simulations, integrating AI with DoD frameworks to provide decision-makers with real-time insights. With a sharp mind for AI/ML and a deep understanding of simulation environments, you thrive in secure, high-stakes settings. You don’t just consume research—you apply it, innovate on it, and deploy it in real-world scenarios that shape national security. Infinity Labs is an innovation-driven company with world-class expertise in the defense and national security sectors. Infinity provides cutting-edge engineering technologies and solutions, including modeling, simulation, and analysis (MS&A), research and development (R&D), cyber operational technology, and training. Our extensive project portfolio covers a vast range of physical scales and applications, from materials and components to subsystems, platforms, and systems of systems. We integrate science and systems as part of our overall solutioning approach and we investigate advanced concepts and technologies with disruptive application potential. The sophistication of our work directly reflects the ingenuity and proficiency of our team. Our workforce shapes the future, and our employees are our greatest asset.

Requirements

• Expertise in machine learning frameworks such as PyTorch, TensorFlow, and JAX.
• Proficiency with reinforcement learning algorithms (PPO, SAC, DQN).
• Hands-on experience integrating AI with modeling and simulation tools (AFSIM, MDASim).
• Strong programming skills in Python and experience with ML libraries like RLlib and Gymnasium.
• Cloud computing experience with AWS or DoD HPC environments.
• Version control proficiency (Git, GitLab, GitHub).
• 10+ years of experience in AI/ML.
• B.S. in Physics, Engineering, or Computer Science (STEM).
• U.S. citizenship is required for consideration, as this job requires U.S. government security clearance
• Active Secret clearance

Nice To Haves

• Experience modifying and extending AFSIM framework capabilities.
• Familiarity with high-level languages such as MATLAB, Simulink, or Julia.
• Deep understanding of component-based architectures in AI/ML applications.
• Demonstrated ability to optimize RL training pipelines for computational efficiency.
• Contributions to published AI/ML research in a defense or security context.
• Active TS/SCI clearance
• Advanced degree preferred.

Responsibilities

• Design, develop, and integrate machine learning models into defense simulations and real-world decision-making systems.
• Apply a broad range of AI/ML techniques, including supervised learning, unsupervised learning, and deep learning to optimize system performance.
• Work with large-scale datasets, developing data pipelines and feature engineering processes to support training and inference.
• Collaborate with DoD civilians, contractors, and stakeholders to align ML solutions with mission objectives.
• Develop high-performance, scalable AI architectures for deployment on cloud and HPC environments.
• Optimize machine learning workflows, ensuring computational efficiency and maintainability in resource-constrained settings.
• Build visualization tools, dashboards, and scripts to analyze model performance and interpret AI-driven insights.
• Stay ahead of the latest advancements in AI/ML, incorporating cutting-edge research into practical applications.
• Engage with the DoD modeling and simulation (M&S) community, presenting findings and collaborating on technical solutions.
• Develop and optimize deep learning architectures, such as transformers, CNNs, and graph-based neural networks, for mission-driven applications.
• Lead the development of custom ML libraries and toolkits to enhance AI capabilities across multiple projects.
• Oversee the design of relational and graph-based database schemas to support large-scale AI-driven analytics.

Benefits

• Comprehensive health benefits including medical, dental, and vision coverage.
• Company-paid disability and life insurance.
• Generous 401(k) plan with guaranteed company contribution.
• Paid time-off options including floating holidays, personal time, parental leave, and community service opportunities.
• Investment in employee growth and development through tuition reimbursement and discounted tuition programs with select colleges.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior