Junior Machine Learning Engineer | AI/ML Gunslinger

Infinity Labs • Colorado Springs, CO

3d • Onsite

About The Position

This position is open for an AI/ML Engineer specializing in reinforcement learning (RL) and defense-related modeling and simulation. Candidates must meet the general requirements below to be considered. 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.

Requirements

• Proficiency with machine learning frameworks such as PyTorch, TensorFlow, and JAX.
• Proficiency with reinforcement learning algorithms (PPO, SAC, DQN).
• 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).
• Bachelor's degree in Physics, Engineering, or Computer Science (STEM)
• Project experience in AI/ML.
• U.S. citizenship is required for consideration, as this job requires U.S. government security clearance

Nice To Haves

• 2+ years of work experience in AI/ML.
• Hands-on experience integrating AI with modeling and simulation tools (AFSIM, MDASim).
• 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.
• Advanced degree in Physics, Engineering, or Computer Science (STEM)
• Active Secret or TS/SCI clearance
• Eligibility to obtain a security clearance

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 encompassing 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.
• Free parking next to the building
• Complimentary snacks and beverages

What This Job Offers

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