Lead ML/AI Engineer

Jobgether; United States; Hybrid; USD 180,000 - 240,000; Full time

Benefits offered by this job

Remote-first work environment

Hybrid option in Nashville

Generous PTO and holidays

Parental leave

Day-1 medical, dental, and vision

Life and disability insurance

Wellness resources and an employer HSA

401(k) retirement plan with employer _

Meaningful healthcare initiatives

Inclusive and collaborative working环境

Job summary

Jobgether is seeking a Lead ML/AI Engineer to own production-grade ML platforms, connecting model outputs to real-world decisions and measuring impact. You will design real-time integrations, establish budgets, and drive CI/CD for ML workflows.

You will partner with data scientists, medical economics, and engineering teams to set standards, review architectures, and mentor staff while traveling up to 10%. This role requires extensive production ML leadership in AWS and Databricks environments.

Qualifications

• 8+ years in ML engineering or software engineering with 4+ years in production ML.
• 2+ years as tech lead or architect of production ML/decisioning systems.
• Experience building prescriptive production systems like recommendation or decisioning platforms.
• Production AWS data & ML services with IaC tools (Terraform, CDK, or CloudFormation).
• Experience with Databricks, PySpark, Delta Lake, MLflow.
• Hands-on with streaming systems (Kinesis/Kafka) and schema management.
• Strong Python and SQL; manage training pipelines, serving, and feature stores.
• Experience implementing CI/CD for ML workloads and monitoring.
• Mentorship and establishing engineering standards; code reviews.
• Travel up to 10%

Responsibilities

• Own the decisioning and feedback layer from model scores to actions.
• Design event-driven real-time integrations between model outputs and operations.
• Define integration contracts and data exchange standards with partner teams.
• Establish cost-to-serve budgets and include as release criteria.
• Set production standards for models, CI/CD, drift monitoring, calibration.
• Build model promotion gates for deployment readiness.
• Own target architecture for AWS-native infra and Databricks lakehouse.
• Collaborate with data scientists and medical/economics teams on methodologies.
• Lead architecture reviews and mentor engineers.

Skills

ML engineering

MLOps

Python

SQL

AWS

Databricks

CI/CD for ML

Streaming systems

Terraform/CDK

Leadership

Education

Bachelor's degree in Computer Science or related field

Equivalent professional experience

Tools

Databricks

PySpark

Delta Lake

MLflow

Kafka

Kinesis

Terraform

CloudFormation

Job description

Accountabilities

As Lead ML/AI Engineer, you will own the engineering foundations that connect model outputs to real-world decisions, ensure reliable feedback and measurement, and establish production standards across the ML/AI platform. Your responsibilities will include:

• Own the decisioning and feedback layer connecting model scores to actions, including action selection and ranking, holdout and experiment assignment, outcome capture, production ROI calculations, and pipelines measuring the impact of actions and channels.
• Lead the engineering implementation, auditability, and reproducibility of production savings calculations in collaboration with medical economics and data science teams.
• Design event-driven, real-time integrations between model outputs and operational systems, including streaming pipelines, versioned schemas, data contracts, replay mechanisms, event ordering, idempotency, and failure handling.
• Define and maintain integration contracts with partner teams to ensure reliable data exchange and system interoperability.
• Establish cost-to-serve standards by defining compute, streaming, inference, and LLM spending budgets at the model and decision levels, making cost a release criterion alongside latency and accuracy.
• Set production standards for ML models, including CI/CD, model registries, alerting, drift monitoring, coverage, and calibration across different cohorts, markets, and payers.
• Build and maintain model promotion gates to ensure that models and ranking systems meet production requirements before deployment.
• Own the target architecture and technical roadmap for the AWS-native infrastructure and Databricks lakehouse, including infrastructure as code, IAM, secure data handling, containerized inference, and managed foundation-model services under PHI controls.
• Define technical direction and production standards for generative AI applications, including LLM pipelines and agentic workflows, with appropriate evaluation, retrieval, guardrails, and cost controls.
• Collaborate with data scientists on modeling and measurement approaches, medical economics teams on savings methodology, and engineering teams on shared pipelines, interfaces, and contracts.
• Establish engineering standards covering code reviews, automated testing, reproducibility, documentation, and release discipline.
• Lead architecture and design reviews, document technical decisions, and mentor engineers and data scientists on system design and production engineering practices.
• Represent the team's technical direction and recommendations to R&D leadership, finance, clinical, and operations stakeholders.

Requirements

The ideal candidate is an experienced ML/AI engineering leader who has built and operated production systems that influence real-world decisions and measure their impact. You should combine strong software and infrastructure expertise with technical leadership, sound engineering judgment, and the ability to collaborate across multidisciplinary teams.

Required Qualifications:

• Bachelor's degree in Computer Science, Engineering, or a related field, or equivalent professional experience.
• At least 8 years of experience in ML engineering, MLOps, or software engineering, including 4+ years operating machine learning systems in production.
• At least 2 years of experience serving as the technical lead or architect of a production ML or decisioning system.
• Demonstrated experience building or operating prescriptive production systems, such as recommendation, decisioning, prioritization, or experimentation platforms that allocate constrained resources against measurable outcomes.
• Production experience with AWS data and machine learning services, as well as infrastructure-as-code tools such as Terraform, CDK, or CloudFormation.
• Production experience with Databricks, PySpark, Delta Lake, MLflow, or equivalent technologies.
• Hands-on expertise designing and operating event-driven or streaming systems using technologies such as Kinesis, Kafka, or equivalent, including schema or contract management.
• Strong Python and SQL skills, with practical experience operating production ML workloads such as training pipelines, model serving, feature systems, or ML platform tooling.
• Experience implementing CI/CD for ML workloads and monitoring models and pipelines, including calibration across heterogeneous populations.
• Proven ability to establish engineering standards, improve team practices through code reviews, and mentor technical colleagues.
• Willingness and ability to travel up to 10% for onsite meetings, team collaboration, and company events.

Preferred Qualifications:

• Experience building and evaluating machine learning models using scikit-learn, XGBoost, or PyTorch.
• Exposure to healthcare economics or value-based care, including savings methodologies, attribution, and payer contract constraints.
• Experience with dbt and lakehouse data modeling.
• Familiarity with healthcare data, including claims, pharmacy, eligibility, prior authorization, and electronic health records (EHR), as well as PHI handling requirements.
• Experience with uplift modeling, bandit algorithms, sequential decisioning, or holdout-based measurement in production.
• Experience with containerized model serving and staged or canary deployment strategies.
• Familiarity with Amazon Bedrock or other managed foundation-model platforms operating under a HIPAA Business Associate Agreement (BAA).
• Experience with agentic frameworks and production retrieval systems.
• Demonstrated experience building or operating LLM-based applications, including evaluation frameworks and guardrails.

Core Competencies:

• Engineering Mentality: Identifies common technical challenges and builds reusable platforms that make future model deployments faster and safer.
• Scientific Approach: Treats system designs as hypotheses, instruments performance, and adapts based on measurable evidence.
• Simplicity: Avoids unnecessary complexity and creates maintainable systems for future users and engineers.
• Ownership: Takes systems from design through deployment and monitoring, remaining accountable for their real-world performance and outcomes.
• Collaboration: Works effectively across data science, data engineering, software engineering, medical economics, clinical, and operations teams to establish clear interfaces and shared standards.
• Communication: Explains technical systems, costs, and business impact in clear terms that clinical and financial stakeholders can use.

Benefits

• Remote-first work environment, with the option of a hybrid arrangement in Nashville.
• Generous paid time off (PTO) and holiday plans.
• Paid parental leave and support for work-life balance, flexibility, and autonomy.
• Medical, dental, and vision insurance starting on Day 1.
• Life and disability insurance.
• Wellness resources and an employer HSA contribution.
• 401(k) retirement plan with employer matching contributions.
• Opportunity to contribute to meaningful healthcare initiatives and improve patient outcomes through technology.
• Inclusive and collaborative working environment that encourages honest feedback, diverse perspectives, and authenticity.
• Commitment to equitable compensation and employee well-being.