Staff Machine Learning Engineer, Oncology foundation Model

July 12

🏄 California – Remote

Although this job is listed in California, this company may be open to hiring remote in other states.

🌽 Illinois – Remote

Although this job is listed in Illinois, this company may be open to hiring remote in other states.

+1 more states

🏄 California – Remote 🌽 Illinois – Remote 🗽 New York – Remote

💵 $170k - $230k / year

⏰ Full Time

🔴 Lead

🤖 Machine Learning Engineer

Apply Now

Receive Emails with Similar Jobs

Tempus AI

WebsiteLinkedInAll Job Openings

Healthcare • Biotechnology • Artificial Intelligence

Tempus AI is an advanced healthcare technology company that specializes in precision medicine through the use of AI and genomic profiling. The company provides innovative solutions across various medical fields, including oncology, cardiology, neurology, and psychiatry. Tempus enables healthcare providers to make informed treatment decisions by integrating real-world data, clinical trial matching, and algorithmic testing to improve patient outcomes and accelerate drug development.

1001 - 5000 employees

Founded 2015

🧬 Biotechnology

🤖 Artificial Intelligence

📋 Description

• Passionate about precision medicine and advancing the healthcare industry? • 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. • 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. • 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.

🎯 Requirements

• 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.

🏖️ Benefits

• incentive compensation • restricted stock units • medical and other benefits depending on the position