AI/ML Engineer
Job Summary
Castalia is seeking an AI/ML Engineer to support a critical customer mission. The role involves developing and implementing automation solutions for machine learning technology integration activities.
Roles And Responsibilities
- 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
- Create scripts and tools to automate repetitive technical tasks
- 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
- 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
- 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)
- Familiarity with streaming data platforms (Kafka, Kinesis)
- Experience supporting federal cybersecurity programs
- DoD 8140 IAT Level III
- CCSP
Physical Requirements/Work Environment
Normal office environment. Travel as required.
Company Description
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
All regularly scheduled employees working at least 30 hours per week are eligible to participate in Castalia Systems' benefit programs. The projected compensation range for this position is $135,000.00 to $145,000.00 (annualized USD).
Disclaimer
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.