AI/ML Automation Engineer – Onsite & Data Pipelines

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

Benefits offered by this job

Medical coverage

Dental coverage

Vision coverage

401k matching

Paid time off

Paid holidays

Professional training opportunities

Pet insurance

Job summary

Castalia Systems is seeking an AI/ML Engineer in Arlington, TX to develop and implement automation solutions for machine learning technology integration. You will focus on agentic workflow development, automated analysis workflows, and integration automation to scale capabilities.

The role requires 5+ years in automation or data engineering, strong Python skills, and experience with cloud-based data pipelines, AI/LLM configurations, and ML pipelines.

Qualifications

• BS in Computer Science, Data Science or related field; or HS Diploma with 7+ years of directly relevant experience.
• 5+ years 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 skills with data manipulation libraries.
• 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.
• Ability to translate business requirements into automated solutions.
• Strong analytical and problem-solving skills.
• Excellent documentation and communication abilities.

Responsibilities

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

ML pipelines

Git/CI-CD

Data modeling

APIs integration

Analytical skills

Data manipulation (Pandas)

Education

BS in CS/Data Science

HS Diploma + 7+ years experience

Tools

Databricks

Airflow

Kubernetes

Docker

Job description

Castalia Systems is seeking an AI/ML Engineer in Arlington, TX to develop and implement automation solutions for machine learning technology integration. You will focus on agentic workflow development, automated analysis workflows, and integration automation to scale capabilities.

The role requires 5+ years in automation or data engineering, strong Python skills, and experience with cloud-based data pipelines, AI/LLM configurations, and ML pipelines.