Senior Machine Learning Engineer | AI​/ML Maestro

Job Title: Machine Learning Engineer | AI/ML Maestro

Location: Colorado Springs, CO

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.

At Infinity Labs, we don't settle for incremental gains—we make bold leaps, pushing the boundaries of what's possible. Our culture thrives on innovation, where creativity fuels action, and inspiration sparks breakthroughs. We don't just embrace calculated risks; we see them as essential steps toward groundbreaking discoveries. We're on a mission to become the epicenter of transformational, multidisciplinary, system-of-systems technology solutions, strategically aligned with the National Defense Strategy.

But our vision isn't just about technology—it's about you. Our structure and benefits are designed to protect your future and unleash your full potential. Join us if you're ready to make a real impact and redefine the future.

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

• 10+ years of experience in AI/ML.
• B.S. in Physics, Engineering, or Computer Science (STEM). Advanced degree preferred.

• Expertise in machine learning frameworks such as PyTorch, Tensor Flow, 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, Git Lab, Git Hub).

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