Viatris

Machine Learning Engineer

Viatris
Apply
1 day ago
Remote, United States or Pittsburgh, PA, USASenior

Responsibilities

  • Build and maintain CI/CD pipelines for machine learning, including automated testing, model deployment, and version control.
  • Deploy machine learning models as scalable APIs and microservices that meet clinical performance and latency requirements.
  • Implement monitoring for model performance, data drift, and production system health.
  • Develop and optimize ETL pipelines that transform FHIR and HL7 healthcare data for model training and inference.
  • Build and maintain feature stores and data layers that keep training and production data consistent.
  • Integrate ML outputs with core healthcare applications in collaboration with backend teams.
  • Write clean, maintainable, documented Python code and participate in code reviews.
  • Package and orchestrate ML workloads with Docker and Kubernetes across environments.
  • Apply security and compliance protocols meeting HIPAA and HITRUST standards.

Requirements

  • Bachelor’s degree or higher in Computer Science, Software Engineering, Data Engineering, or a related field.
  • At least 6 years of professional software engineering or data engineering experience.
  • At least 5 years of Python experience and familiarity with SQL.
  • At least 2 years of experience working in machine learning production environments.
  • At least 5 years of experience with AWS and Docker-based containerization.
  • At least 3 years of experience working with Java.
  • At least 3 years of experience with PyTorch or Scikit-learn and MLOps tools such as Airflow, Prefect, BentoML, or Kubeflow.
  • At least 4 years of experience with data-processing frameworks such as Pandas, Spark, or dbt.
  • Familiarity with deploying large language models or using LangChain is preferred.
  • Experience in a regulated environment and understanding of API design and microservices architecture are preferred.

Benefits

  • Remote work arrangement.
  • May include up to 10% domestic travel.
  • Company-paid and coordinated on-site onboarding travel for applicable new hires with start dates of October 1, 2025 or later.
  • Benefits may include medical, dental and vision coverage, health savings accounts, flexible spending accounts, disability benefits, life insurance, voluntary benefits, paid absences, and retirement benefits.

Tech Stack

Apache AirflowApache SparkAWSdbtDockerGitJavaKubernetesMLflowPandasPythonPyTorchscikit-learnSQL

Categories

Data EngineeringML Engineering
Contact me