AI/ML Engineer

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.

Compensation

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 insurance to ensure your furry friends are cared for too. All regularly scheduled employees working at least 30 hours per week are eligible to participate in Castalia Systems’ benefit programs. Individuals that do not meet the threshold are only eligible for select offerings, not inclusive of health benefits.

Salary at Castalia Systems is determined by various factors, including but not limited to location, position knowledge, skills, competencies, and experience, as well as contract-specific affordability and organizational requirements. The projected compensation range for this position is $135,000.00 to $145,000.00 (annualized USD). The estimate displayed represents the typical salary range for this position and is just one component of Castalia Systems’ total compensation package for employees.

Castalia Systems is an equal employment opportunity and affirmative action employer and strives to comply with all applicable laws prohibiting discrimination based on race, color, creed, sex, sexual orientation, age, national origin, or ancestry, physical or mental disability, veteran status, marital status, HIV-positive status, as well as any other category protected by federal, state, or local laws. All such discrimination is unlawful, and all persons involved in the operations of the company are prohibited from engaging in this type of conduct.