AI/ML Engineer

Castalia Systems

$135K — $145K *
Information Technology
5 - 7 years of experience
Job Overview by Ladders

Qualifications

  • BS in Computer Science, Data Science, or related field; HS Diploma with 7+ years relevant experience accepted.
  • 5+ years in automation engineering, data engineering, or software development.
  • Experience designing data ingestion pipelines on AWS.
  • Familiarity with AI/LLM configuration, tuning, and integration.
  • Strong Python programming skills with data manipulation libraries.
  • Experience with Amazon Bedrock and API integration.
  • Good understanding of database systems and data modeling.

Responsibilities

  • Develop automated data pipelines for machine learning initiatives.
  • Create workflows for malware sample processing and analysis.
  • Implement AI/LLM integrations with Databricks and cloud services.
  • Support development of the Automated Malware Analysis platform.
  • Develop scripts to automate technical tasks and processes.
  • Create automation for data transformation and normalization.
  • Collaborate with analysts to identify new automation opportunities.

Benefits

  • Medical, dental, and vision coverage.
  • 401k matching and generous PTO.
  • Paid holidays and professional development opportunities.
  • Pet insurance for employees' furry friends.
Full Job Description
Job Type: Full- Time

Workplace Type: Onsite

Clearance: Secret with ability to get a DHS Suitability

Must be a U.S. Citizen

Job Summary

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.

Roles and Responsibilities

A qualified candidate will perform the following duties and responsibilities, but are not limited to:
  • 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

Required Qualifications
  • 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

Preferred Qualifications
  • 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

Physical Requirements/Work Environment
  • Normal office environment.

Travel
  • As required

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.

function load() { var urllocation = location.href; //find url parameter if(urllocation.indexOf("#") == -1 || urllocation.indexOf("#job") > -1){ window.location.hash="job"; } }

Similar Jobs

More Jobs at Castalia Systems

More Information Technology Jobs

Find similar AI/ML Engineer jobs: