Staff Machine Learning Engineer

Tempus AI; New York (NY); On-site; USD 190,000 - 230,000; Full time

Job summary

A leading company in precision medicine is seeking a Staff Machine Learning Engineer to design and optimize data infrastructure for advanced AI models. This role focuses on managing data workflows and enhancing patient care through innovative AI applications. Ideal candidates will have extensive experience in data engineering and a strong academic background in AI. Join a dynamic team at the forefront of healthcare technology.

Qualifications

• 8+ years of industry experience in large-scale data pipelines.
• Experience with TBs+ heterogeneous datasets.
• Proficiency in programming languages like Python.

Responsibilities

• Architect and build data processing workflows for multimodal training data.
• Develop strategies for efficient data ingestion from diverse sources.
• Collaborate with teams to optimize cloud-native services.

Skills

Leadership

Collaboration

Communication

Data Engineering

Problem Solving

Python

Education

Master's degree in Computer Science

Tools

Apache Spark

Ray

Dask

MosaicML Streaming

Hugging Face Datasets

MLflow

Kubeflow

SageMaker Pipelines

Job description

Join to apply for the Staff Machine Learning Engineer role at Tempus AI

1 week ago Be among the first 25 applicants

Join to apply for the Staff Machine Learning Engineer role at Tempus AI

Get AI-powered advice on this job and more exclusive features.

Passionate about precision medicine and advancing the healthcare industry?

Recent advancements in underlying technology have finally made it possible for AI to impact clinical care in a meaningful way. Tempus' proprietary platform connects an entire ecosystem of real-world evidence to deliver real-time, actionable insights to physicians, providing critical information about the right treatments for the right patients, at the right time.

What You’ll Do

We are seeking an experienced and highly skilled Staff Machine Learning Engineer with deep expertise in large-scale multimodal model systems engineering to join our dynamic AI team. You will play a pivotal role in designing, building, and optimizing the foundational data infrastructure that powers Tempus's most advanced generative AI models. Your work will directly enable the training and deployment of robust, production-ready multimodal systems that analyze complex data types (like genomics, pathology images, radiology scans, and clinical notes) to improve patient care, optimize clinical workflows, and accelerate life-saving medical research. This is a critical, high-impact position for driving the practical application of cutting-edge AI to revolutionize healthcare.

Focus

Your primary focus will be to architect, build, and maintain the critical data infrastructure supporting our large multimodal generative models. This includes managing the entire lifecycle of vast datasets – from ingestion and processing of diverse training data to the integration and retrieval of extensive knowledge sources used to augment model capabilities. You will be building the data backbone that enables our AI to learn from Tempus's rich real-world evidence.

Key Responsibilities

As a technical leader in this space, you will be:

• Architect and build sophisticated data processing workflows responsible for ingesting, processing, and preparing multimodal training data that seamlessly integrate with large-scale distributed ML training frameworks and infrastructure (GPU clusters).
• Develop strategies for efficient, compliant data ingestion from diverse sources, including internal databases, third-party APIs, public biomedical datasets, and Tempus's proprietary data ecosystem.
• Utilize, optimize, and contribute to frameworks specialized for large-scale ML data loading and streaming (e.g., MosaicML Streaming, Ray Data, HF Datasets).
• Collaborate closely with infrastructure and platform teams to leverage and optimize cloud-native services (primarily GCP) for performance, cost-efficiency, and security.
• Engineer efficient connectors and data loaders for accessing and processing information from diverse knowledge sources, such as knowledge graphs, internal structured databases, biomedical literature repositories (e.g., PubMed), and curated ontologies.
• Optimize data storage for efficient large scale training training and knowledge access.
• Orchestrate, monitor, and troubleshoot complex data workflows using tools like Airflow, Kubeflow Pipelines..
• Establish robust monitoring, logging, and alerting systems for data pipeline health, data drift detection, and data quality assurance, providing feedback loops for continuous improvement.
• Analyze and optimize data I/O performance bottlenecks considering storage systems, network bandwidth and compute resources.
• Actively manage and seek optimizations for the costs associated with storing and processing massive datasets in the cloud.
  
  

Required Skills And Experience

• Master's degree in Computer Science, Artificial Intelligence, Software Engineering, or a related field. A strong academic background with a focus on AI data engineering.
• Proven track record (8+ years of industry experience) in designing, building, and operating large-scale data pipelines and infrastructure in a production environment.
• Strong experience working with massive, heterogeneous datasets (TBs+) and modern distributed data processing tools and frameworks such as Apache Spark, Ray, or Dask.
• Strong, hands-on experience with tools and libraries specifically designed for large-scale ML data handling, such as Hugging Face Datasets, MosaicML Streaming, or similar frameworks (e.g., WebDataset, Petastorm). Experience with MLOps tools and platforms (e.g., MLflow, Kubeflow, SageMaker Pipelines).
• Understanding of the data challenges specific to training large models (Foundation Models, LLMs, Multimodal Models).
• Proficiency in programming languages like Python and experience with modern distributed data processing tools and frameworks.
• Leadership and collaboration:
• Proven ability to bring thought leadership to the product and engineering teams, influencing technical direction and data strategy.
• Experience mentoring junior engineers and collaborating effectively with cross-functional teams (Research Scientists, ML Engineers, Platform Engineers, Product Managers, Clinicians).
• Excellent communication skills, capable of explaining complex technical concepts to diverse audiences.
• Strong bias-to-action and ability to thrive in a fast-paced, dynamic research and development environment.
• A pragmatic approach focused on delivering rapid, iterative, and measurable progress towards impactful goals.
  

Preferred Qualifications

• Advanced degree (PhD) in Computer Science, Engineering, Bioinformatics, or a related field.
• Contributions to relevant open-source projects.
• Direct experience working with clinical or biological data (EHR, genomics, medical imaging).
  
  

California Pay Range

New York Pay Range - $190,000 - $230,000 USD

California Pay Range - $190,000 - $230,000 USD

Illinois Pay Range - $170,000 - $210,000 USD

Remote - USA Range - $170,000 - $210,000 USD

The expected salary range above is applicable if the role is performed from California and may vary for other locations (Colorado, Illinois, New York). Actual salary may vary based on qualifications and experience. Tempus offers a full range of benefits, which may include incentive compensation, restricted stock units, medical and other benefits depending on the position.

Additionally, for remote roles open to individuals in unincorporated Los Angeles – including remote roles- Tempus reasonably believes that criminal history may have a direct, adverse and negative relationship on the following job duties, potentially resulting in the withdrawal of the conditional offer of employment: engaging positively with customers and other employees; accessing confidential information, including intellectual property, trade secrets, and protected health information; and appropriately handling such information in accordance with legal and ethical standards. Qualified applicants with arrest or conviction records will be considered for employment in accordance with applicable law, including the Los Angeles County Fair Chance Ordinance for Employers and the California Fair Chance Act.

We are an equal opportunity employer. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Job function

  Engineering and Information Technology

Referrals increase your chances of interviewing at Tempus AI by 2x

Get notified about new Machine Learning Engineer jobs in New York, NY.

New York, NY $100,000.00-$200,000.00 1 year ago

New York, NY $56.25-$173,000.00 2 weeks ago

New York, NY $90,000.00-$140,000.00 11 months ago

Brooklyn, NY $150,000.00-$200,000.00 3 weeks ago

Want to work with us, but don't see the right job listed?

New York, NY $170,000.00-$250,000.00 1 week ago

Jr. Software Development Engineer - New York, NY

New York, NY $150,000.00-$200,000.00 3 months ago

New York, NY $140,000.00-$200,000.00 3 months ago

Software Engineer, Search Infrastructure

New York, NY $220,000.00-$260,000.00 4 days ago

New York, NY $150,000.00-$200,000.00 3 weeks ago

New York, NY $120,000.00-$180,000.00 2 months ago

New York, NY $175,000.00-$250,000.00 3 days ago

New York, NY $140,000.00-$170,000.00 1 month ago

New York, NY $150,000.00-$180,000.00 2 weeks ago

New York, NY $140,000.00-$170,000.00 1 week ago

New York, NY $100,000.00-$200,000.00 1 year ago

New York, NY $140,000.00-$185,000.00 2 days ago

New York, NY $90,000.00-$120,000.00 3 weeks ago

New York, NY $46.63-$134,000.00 6 days ago

Software Engineer, Application Foundations Team

New York, NY $165,000.00-$165,000.00 1 year ago

New York, NY $70,000.00-$120,000.00 1 month ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.