AI​/ML Engineer

Job Type: Full
- Time

Workplace Type: Onsite

Clearance: Secret with ability to get a DHS Suitability

Must be a U.S. Citizen

Castalia is seeking a AI/ML Engineer to support this critical customer mission. The AI/ML Engineer will develop and implement automation solutions to support machine learning technology integration activities, focusing on agentic workflow development, automated analysis workflows, and integration automation. This role addresses manual workflow pain points and enables scaling of technology capabilities through intelligent automation.

• Develop automated data pipelines for machine learning technology integration initiatives
• Create automation workflows for malware sample collection, detonation, and taxonomy
• Implement AI/LLM integrations with Databricks and other cloud services
• Implement data integration automation between pilot and production systems
• Support Automated Malware Analysis (AMA) platform development and enhancement
• Develop scripts and tools to automate repetitive technical tasks
• Create data transformation and normalization automation
• Support AI/ML data preparation and feature engineering automation
• Develop monitoring and alerting automation for integrated systems
• Implement API integrations for data sharing and workflow automation
• Document automation solutions and maintain code repositories
• Support scaling automation from pilot to production environments
• Optimize automation performance and efficiency
• Collaborate with analysts to identify automation opportunities
• Ensure automation solutions maintain data quality and integrity
• Support continuous improvement of automated capabilities

• BS in Computer Science, Data Science, or related degree, or HS Diploma and 7+ years of directly relevant experience.
• 5+ years of experience in automation engineering, data engineering, or software development
• Experience with data ingestion pipeline design on AWS
• Experience with AI/LLM configuration, tuning, and integration
• Experience with Machine Learning (ML) pipelines
• Strong programming skills in Python, with experience in data manipulation libraries
• Experience Amazon Bedrock (model access, endpoints, API Integration)
• Knowledge of automation frameworks and orchestration tools
• Experience with APIs and integration technologies
• Understanding of database systems and data modeling
• Familiarity with version control (Git) and CI/CD practices
• Strong problem-solving and analytical skills
• Ability to translate business requirements into automated solutions
• Strong analytical and problem-solving skills
• Excellent documentation and communication abilities

• TS/SCI clearance
• Experience with malware analysis automation or cybersecurity tools
• Knowledge of AI/ML pipelines and data preparation
• Experience with Databricks and Databricks AI
• Experience with workflow orchestration tools (Airflow, Prefect)
• Background in cybersecurity or threat intelligence automation
• Experience with containerization and orchestration (Docker, Kubernetes)
• Knowledge of streaming data platforms (Kafka, Kinesis)
• Familiarity with big data technologies (Spark, Hadoop)
• Experience supporting federal cybersecurity programs
• DoD 8140 IAT Level III
• CSSLP

• Normal office environment.

• As required

Castalia Systems is a proven business partner providing mission critical solutions to the Federal Government. We provide cutting edge solutions from Securing and Managing Data to Systems Engineering and Development. Castalia Systems is a pioneer in Artificial Intelligence Design and Application.

With our vast knowledge of our customers' needs and relevant technology, our team is able to bring successful solutions to every mission. We are one-upping our competitors by providing premium IT solutions and platforms with cutting-edge technology so it's so evident when you compare us with anyone.

At Castalia Systems, we provide you with opportunities and choices and support your total well‑being. Our benefits include:
Medical, dental, vision coverage, 401k matching, generous PTO, paid holidays, professional training opportunities, and even pet…