AI / Machine Learning Engineer - Senior

Falls Technology is hiring a Senior AI / Machine Learning Engineer to train machine learning models and incorporate pre-trained models into production data workflows for a federal national-security mission. The role focuses on building reliable AI workflows using AWS ETL pipelines and implementing software components in Python, Java, and C++ with PyTorch or TensorFlow.

Key Responsibilities

• Automate workflows and business processes using AI to support operational needs.
• Train models and integrate pre-trained models into production data flows.
• Design data structures and data models, and contribute to software architecture.
• Develop AI and ML software components using Python, Java, and C++, leveraging PyTorch or TensorFlow.
• Build and support ETL pipelines on AWS to move and transform data for downstream AI capabilities.

Required Qualifications

• Active TS/SCI with Polygraph.
• Hands-on experience with PyTorch or TensorFlow.
• Proficiency in Python, Java, C++, and Conda.
• Strong foundation in probability, statistics, and algorithms.
• Experience with data structures, data modeling, and software architecture.
• Ability to train and integrate ML models into data-driven systems.

Technologies and Tools

• PyTorch, TensorFlow
• Python, Java, C++
• Conda
• ETL, AWS

Location

Chantilly, VA (onsite).

Compensation

USD 175,000 per year.

Benefits

• 9% retirement contribution
• 100% of family medical premiums paid
• 4 weeks of PTO

Nice to Have

• Experience with secure AI on sensitive data.
• Experience leading AI/ML initiatives.
• Additional background in ETL pipelines and AWS.
• Hugging Face / LLM experience.