Staff Machine Learning Researcher - Biological Foundational Models jobs in United States
cer-icon
Apply on Employer Site
company-logo

Somite AI · 2 months ago

Staff Machine Learning Researcher - Biological Foundational Models

Somite AI is a venture-backed company aiming to become the OpenAI of stem cell biology, developing AI foundation models for human stem cells to drive novel cell therapies at scale. As a Senior Machine Learning Researcher, you will develop foundational models for single-cell RNA sequencing data and collaborate with computational biologists to create cutting-edge AI solutions.

Artificial Intelligence (AI)Biotechnology
check
H1B Sponsor Likelynote

Responsibilities

Design, train, and optimize foundational models for single-cell RNA sequencing and other high-dimensional omics data
Collaborate closely with computational biologists to ensure models produce biologically meaningful, interpretable outputs, engaging deeply with biological questions and research
Design and implement novel deep learning architectures, including transformer-based models tailored to biological data
Build scalable, distributed pipelines for training and inference across trillion-token biological datasets
Provide mentorship and technical guidance to other machine learning researchers on the team

Qualification

Machine LearningDeep LearningPythonTransformer-based ModelsLarge-scale Data AnalysisSingle-cell RNA SequencingComputational BiologyIndependent ResearchBiological ModelingReinforcement LearningCollaborationMentorship

Required

6+ years of experience in machine learning, transformer-based deep learning, and large-scale data analysis, preferably in biological applications
Master's degree or higher in Computer Science, Artificial Intelligence, Computational Biology, or a related field
Strong proficiency in Python and deep learning frameworks such as PyTorch or TensorFlow
Proven experience building and optimizing large-scale models, including transformer-based architectures
Demonstrated track record of independent research and end-to-end model development, from prototyping to production
Passion for AI-driven biology and its potential to transform healthcare

Preferred

Strongly preferred: PhD with a strong publication record in top machine learning and/or computational biology journals
Strongly preferred: extensive background in biological research and biological modeling, ideally with single-cell RNA sequencing data
Experience with reinforcement learning (e.g., RLHF, PPO), diffusion models, or multi-modal architectures applied to structured, high-dimensional data
Experience in startups or fast-paced environments, with a self-directed, proactive work style

Benefits

Take a technical leadership role with a mission-driven company with the potential to significantly impact the lives of millions
Work alongside a talented and passionate team at the forefront of AI and cellular biology
Contribute to the development of groundbreaking therapies that address significant unmet medical needs
Enjoy a competitive salary and benefits package, including flexible work options

Company

Somite AI

twittertwittertwitter
company-logo
Somite.ai is developing AI foundation models for human stem cells to drive novel cell therapies at scale.

H1B Sponsorship

Somite 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
2025 (5)
2024 (2)

Funding

Current Stage
Early Stage
Total Funding
$104.79M
Key Investors
AMD VenturesKhosla VenturesTechAviv
2025-08-20Series Unknown· $47.5M
2025-05-13Series A· $47M
2024-04-10Pre Seed· $10.29M

Leadership Team

leader-logo
Micha Y. Breakstone
Co-founder and CEO
linkedin
leader-logo
Allon Klein
Co-Founder
linkedin

Recent News

Company data provided by crunchbase