AI Engineer

The role

You will work across the full ML lifecycle to transform real-world data into production-grade models. You will design, train, and deploy ML models and build scalable data pipelines for defense applications.

What you will do

• Design, train, fine-tune, and deploy ML models
• Own the lifecycle from data exploration to production monitoring
• Build scalable data pipelines for structured and unstructured data
• Develop reliable training and inference systems for performance and edge deployment
• Integrate models into robust, scalable production systems
• Prototype and benchmark new modeling approaches

Required qualifications

• Bachelor's degree (B.Sc.) in Computer Science
• Strong software engineering fundamentals
• Solid understanding of machine learning principles
• Hands-on experience with ML frameworks (PyTorch, TensorFlow, Hugging Face)
• Experience working with real-world datasets
• Strong communication and collaboration skills
• Willingness to travel up to 15% to engage with DoW partners
• Must be eligible to obtain and maintain a U.S. security clearance

Nice to have

• Master's degree (M.Sc.) in Computer Science, Machine Learning, or related field
• Experience with edge-deployable or offline ML systems
• Experience optimizing models for latency and compute constraints
• Familiarity with distributed training or large-scale data processing
• Exposure to geospatial, multimodal, or reasoning systems
• Experience with containerization and orchestration tools

Benefits

• Competitive salary and equity
• Opportunity to help build a category-defining defense tech company
• Full health benefits
• 401(k)

About the company

Pytho builds AI for defense to ensure warfighters are prepared for every challenge. Their software is deployed and in use with DoW partners.