Machine Learning Engineer - Mid

About the role

SOSi is seeking a Mid-Level Machine Learning Engineer to support the development, integration, and deployment of machine learning capabilities for a government customer in Huntsville, Alabama.

The selected candidate will contribute to an MLOps environment supporting national security missions and large-scale analytical workloads. The position focuses on transforming machine learning prototypes into reliable, maintainable, and production-ready capabilities.

The ideal candidate has practical experience developing machine learning models, writing production-quality Python, and integrating models into established software systems. Candidates should understand the full machine learning lifecycle, including data preparation, experimentation, evaluation, deployment, monitoring, and continuous improvement.

The engineer will work alongside data scientists, data engineers, and software developers in an Agile environment to deliver solutions aligned with mission requirements.

The team works with technologies including Python, PyTorch, scikit-learn, MLflow, PostgreSQL, Docker, Linux, GitLab, Jupyter Notebooks, and supporting MLOps infrastructure.

Essential Job Duties

Machine Learning Development

• Develop, train, evaluate, and improve machine learning models for operational applications.
• Apply supervised and unsupervised learning techniques to structured and unstructured datasets.
• Perform feature engineering, data preprocessing, and dataset validation.
• Select appropriate algorithms and evaluation metrics based on project requirements.
• Conduct experiments, compare model performance, and document technical findings.
• Identify and address model performance issues, including overfitting, data leakage, and prediction errors.

Production ML Engineering

• Convert research prototypes and notebooks into maintainable Python modules and services.
• Integrate trained models into existing applications, APIs, and processing workflows.
• Package and deploy model inference services using containerized environments.
• Implement consistent model loading, configuration management, and error handling.
• Support model versioning, experiment tracking, and reproducible deployments.
• Investigate and resolve model inference, integration, and performance issues.
• Contribute to automated model validation and deployment workflows.

Software Development and Quality Assurance

• Write clean, modular, testable, and maintainable Python code.
• Develop unit tests for preprocessing, feature engineering, and model-related functionality.
• Develop integration tests validating model interactions with dependent systems.
• Implement smoke tests confirming deployed services operate as expected.
• Participate in peer code reviews and address identified defects.
• Use Git-based development workflows and established CI/CD practices.
• Maintain technical documentation covering implementation, testing, and operational procedures.

Data Integration and MLOps

• Collaborate with data engineers to establish reliable training and inference datasets.
• Implement validation checks for dataset quality, schema consistency, and missing values.
• Work with PostgreSQL and other approved data sources.
• Use MLflow or comparable tools to manage experiments and model artifacts.
• Support model registry workflows, version promotion, and rollback procedures.
• Monitor model behavior, resource utilization, and inference performance.
• Identify data drift, performance degradation, and operational anomalies.
• Support reproducible model training and deployment processes.

Collaboration and Delivery

• Complete assigned development tasks within agreed sprint objectives.
• Communicate implementation progress, technical blockers, and delivery risks.
• Provide realistic estimates for assigned engineering efforts.
• Coordinate interface requirements with data engineering and software development teams.
• Follow established architecture, coding, testing, and documentation standards.
• Identify proposed scope changes before implementing unapproved functionality.
• Participate in technical discussions, planning sessions, and code reviews.

Minimum Requirements

• Active Top Secret clearance with SCI eligibility.
• Bachelor's degree in Computer Science, Engineering, Mathematics, Statistics, Data Science, or a related quantitative discipline.
• 3-5 years of relevant experience in machine learning engineering, software development, or applied data science.
• Proficiency in Python and common machine learning libraries.
• Practical experience training, evaluating, and deploying machine learning models.
• Experience with machine learning frameworks such as PyTorch, TensorFlow, or scikit-learn.
• Experience using SQL and relational databases.
• Working knowledge of Linux development environments and command-line tools.
• Experience with Git, collaborative development, and pull request workflows.
• Familiarity with Docker and containerized application deployment.
• Understanding of automated testing, including unit and integration testing.
• Knowledge of REST APIs and integration between software components.
• Understanding of CI/CD principles and Agile development methodologies.
• Ability to independently implement, troubleshoot, and deliver assigned engineering tasks.

Preferred Qualifications

• Experience deploying and maintaining machine learning models in production.
• Experience with MLflow or similar experiment tracking and model registry platforms.
• Familiarity with Triton Inference Server or comparable inference serving platforms.
• Experience developing computer vision, NLP, classification, or anomaly detection solutions.
• Experience integrating machine learning models into ETL/ELT pipelines.
• Familiarity with Apache Airflow or similar workflow orchestration tools.
• Experience with PostgreSQL, large-scale datasets, or distributed processing.
• Understanding of model monitoring, data drift, reproducibility, and rollback strategies.
• Familiarity with GitLab CI/CD, automated validation, and container registries.
• Experience developing solutions in restricted, offline, or on-premises computing environments.
• Experience supporting DoD or Intelligence Community missions.
• Exposure to high-performance computing and multi-INT analytical workloads.

Additional Information: Work Environment

• Office-based position in Huntsville, Alabama.
• Collaboration with multidisciplinary engineering and government stakeholder teams.
• Work involving mission-sensitive data and secure computing environments.
• Occasional remote work or travel may be available based on program requirements.

Working at SOSi

All interested individuals will receive consideration and will not be discriminated against for any reason.