Infinity Labs is looking for a Senior AI/ML Engineer to design, develop, and integrate machine learning models into defense simulations and real-world decision-making systems, focusing on reinforcement learning and defense-related modeling and simulation.
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).
- Experience modifying and extending AFSIM framework capabilities.
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.
- 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.
- Develop and optimize deep learning architectures, such as transformers, CNNs, and graph-based neural networks, for mission-driven applications.
Other
- 10+ years of experience in AI/ML.
- US Citizenship is REQUIRED.
- Must have -- Secret level clearance.
- Nice to have -- Top Secret level clearance and current SCI access.
- Must be able to remain in a stationary position and work on a computer for prolonged periods.
