Tempus AI · 1 month ago
Staff Machine Learning Scientist, Oncology Foundation Model
United States
Full-time
Remote
Lead/Staff
$220K/yr - $260K/yrTempus AI is a company focused on precision medicine and advancing the healthcare industry through AI technology. They are seeking a Staff Machine Learning Scientist to contribute to the design and architecture of large multimodal models (LMMs) and build infrastructure for training at scale.
Artificial Intelligence (AI)BiotechnologyHealth CareMachine LearningMedicalPrecision Medicine
Responsibilities
Design and definition of the architecture of the LMMs, considering different fusion strategies and modality-specific processing
Implement, refine, benchmark and optimize model architectures using deep learning frameworks such as PyTorch or TensorFlow
Develop and manage the end-to-end training pipelines, including data loading, preprocessing, and model training. Architect and deploy distributed training workflows, optimizing for performance across cloud GPU fleets
Implement distributed training strategies to handle large-scale datasets and models
Design and implement methods to fuse knowledge with the multimodal representations within the LMM
Experiment with different approaches to enhance the model's understanding and reasoning abilities through knowledge integration
Monitor and debug training processes, identifying and resolving performance bottlenecks
Collaborate with the knowledge integration engineer to ensure the architecture can accommodate knowledge injection mechanisms
Qualification
Required
Deep understanding of deep learning principles and architectures (especially transformers)
Extensive experience with multimodal machine learning concepts and techniques (for example, different fusion methods for text and images)
Solid understanding of optimization techniques for large-scale models
Strong proficiency in Python and deep learning frameworks (PyTorch/TensorFlow) and model management libraries like HF Transformers
Experience with training large multimodal models with distributed training frameworks (for example, Horovod, MosaicML) and GPU fleet management
Strong understanding of knowledge representation concepts (for example, knowledge graphs, ontologies)
Experience with distributed training frameworks and cloud computing platforms (for example, GCP, Azure)
Benefits
Incentive compensation
Restricted stock units
Medical and other benefits depending on the position
Company
Tempus AI
Tempus is making precision medicine a reality by applying AI in healthcare, deriving insights from our expansive library of clinical data and molecular data.
Founded in 2015
1001-5000 employees
H1B Sponsorship
Tempus AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2021 (3)
Funding
Current Stage
Public CompanyTotal Funding
$2.29BKey Investors
Ares ManagementGoogleBaillie Gifford
2025-06-30Post Ipo Debt· $650M
2025-02-19Post Ipo Debt· $300M
2024-06-14IPO
Leadership Team
Eric Lefkofsky
Founder and CEO
Shane Colley
CTO
Recent News
2026-01-06
Company data provided by crunchbase