Staff Machine Learning Engineer, Applied Science at Hippocratic AI

Palo Alto, California, United States

Hippocratic AI Logo
Not SpecifiedCompensation
Senior (5 to 8 years), Expert & Leadership (9+ years)Experience Level
Full TimeJob Type
UnknownVisa
HealthcareIndustries

Requirements

  • Bachelor’s or Master’s degree in Computer Science, Engineering, or a related technical field; PhD is a plus
  • 7+ years of industry experience building and deploying ML systems, ideally with focus on LLMs or deep learning
  • Strong software engineering skills and experience developing in Python
  • Deep experience with ML frameworks like PyTorch or TensorFlow
  • Familiarity with distributed training frameworks (e.g., DeepSpeed, FSDP, Horovod)
  • Proven experience designing, implementing, and scaling ML pipelines for large models
  • Experience with cloud platforms (e.g., AWS, GCP, Azure) and container orchestration (e.g., Kubernetes, Docker)
  • Exposure to healthcare, clinical, or life sciences data is a strong plus

Responsibilities

  • Designing scalable, high-performance infrastructure for LLM training, fine-tuning, and inference
  • Building ML pipelines and tooling for experimentation, evaluation, and deployment
  • Optimizing model performance and efficiency in production environments
  • Collaborating cross-functionally to integrate ML solutions into end-user applications
  • Maintaining compliance with healthcare standards of safety, privacy, and reliability
  • Contributing to technical decision-making and defining the roadmap for scalable, safe ML systems

Skills

Python
PyTorch
TensorFlow
DeepSpeed
FSDP
Horovod
Kubernetes
Docker
AWS
GCP
Azure

Hippocratic AI

About Hippocratic AI

N/AHeadquarters
N/AYear Founded
N/ACompany Stage

Land your dream remote job 3x faster with AI