Senior Machine Learning Engineer | AI/ML Maestro at pmX Group

Huntsville, Alabama, United States -

Full Time

Start Date

Immediate

Expiry Date

11 Aug, 26

Salary

0.0

Posted On

13 May, 26

Experience

10 year(s) or above

Remote Job

Yes

Telecommute

Yes

Sponsor Visa

Skills

Reinforcement Learning, PyTorch, TensorFlow, JAX, Python, AFSIM, MDASim, RLlib, Gymnasium, AWS, Git, Deep Learning, Modeling and Simulation, Feature Engineering, HPC Environments, Graph-based Neural Networks

Industry

Description

Job Title: Machine Learning Engineer | AI/ML Maestro Location: Huntsville, AL 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. Who You Are: 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. Who We Are: 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. What You’ll Do: * 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. Minimum Qualifications/What You'll Bring: * 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). Education/Credentials: * 10+ years of experience in AI/ML. * B.S. in Physics, Engineering, or Computer Science (STEM). Advanced degree preferred. * U.S. citizenship is required for consideration, as this job requires U.S. government security clearance * Required - Active Secret clearance * Highly Desired/Preferred - Active TS/SCI 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. Physical Demands: * Must be able to remain in a stationary position and work on a computer for prolonged periods. This job description provides a high-level review of the responsibilities of the position and is not intended to be a complete list of all responsibilities, duties or skills that may be required for the job. The job description is subject to review and change at any time and other job-related duties or requirements may be assigned as necessary. What We Offer: As a nationally recognized Great Place to Work® and Military Times® Best for Vets: 2025 Employer, we prioritize employee well-being and experience, fostering a culture that values dedicated teams and committed individuals. Our premier total compensation package for eligible employees includes: * 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. Infinity Labs is an equal opportunity employer, including disability/vets.

Responsibilities

Design and integrate machine learning models into defense simulations and real-world decision-making systems. Develop scalable AI architectures and data pipelines to optimize system performance within DoD frameworks.