Onsite AI/ML Automation Engineer – Data Pipelines & AI

Castalia Systems; Arlington (VA); On-site; USD 135,000 - 145,000; Full time

Benefits offered by this job

Medical benefits

Dental benefits

401k matching

Paid time off

Pet insurance

Job summary

Castalia Systems is seeking an AI/ML Engineer to develop automation solutions for machine learning integration, focusing on agentic workflows and scalable analytics. The role addresses data pipelines, automated analysis, and cloud-based integration to support critical missions.

Responsibilities include building automated data flows, malware workflow automation, and AI/LLM integrations with cloud services, while ensuring data quality and maintainability across pilot to production environments.

Qualifications

• BS in Computer Science, Data Science, or related field or HS Diploma with 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 ML pipelines.
• Strong Python programming and data manipulation skills.
• Experience with 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 Git and CI/CD practices.
• Strong problem-solving and analytical skills.
• Ability to translate business requirements into automated solutions.
• Excellent documentation and communication abilities.

Responsibilities

• Develop automated data pipelines for ML 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.

Skills

Python

AWS data pipelines

AI/LLM integration

Databricks

Git

CI/CD

Data modeling

APIs

Education

Bachelor's in CS or related

Tools

Databricks

Airflow

Kubernetes

Docker

Kafka

Kinesis

Spark

Hadoop

Job description

Castalia Systems is seeking an AI/ML Engineer to develop automation solutions for machine learning integration, focusing on agentic workflows and scalable analytics. The role addresses data pipelines, automated analysis, and cloud-based integration to support critical missions.

Responsibilities include building automated data flows, malware workflow automation, and AI/LLM integrations with cloud services, while ensuring data quality and maintainability across pilot to production environments.