Axiom Bio · 2 months ago
Machine Learning
SF Global HQ
Full-time
Onsite
Mid, Senior LevelAxiombio is focused on building accurate AI systems to replace lab and animal toxicity experiments. The role involves defining and leading the development of machine learning systems, conducting research on novel models, and scaling large models, all while fostering an entrepreneurial spirit and creating value for scientists.
Artificial Intelligence (AI)Biotechnology
Responsibilities
Define the core end-to-end ML/AI system: wetlab data creation and data cleaning/processing; model architecture, training, and inference; compute infrastructure and model deployment
Lead the research & development of novel models for understanding the relationship between chemistry and biology
Architect and scale large models pretrained on paired chemistry and biological images
Applied research on optimizing, aggregating, and pooling embeddings
Become a research leader in novel and underserved areas: molecular graph representations, contrastive learning for chem/bio, semi-supervised learning on biological images, generative diffusion for biology
Grow as an entrepreneur as well as an engineer by building things from 0 to 1 while creating incredible value for scientists
Ship insanely great technology + products
Qualification
Required
Has done at least one piece of work, in industry or in academia, that shows exceptional machine learning talent
Strong engineering abilities (writing Pytorch NN code, working with data in numpy / pandas, interacting with training and inference pipelines, and writing more general software in Python and interacting with the cloud)
Extremely high potential to become a leader in the field of ML/AI research
Demonstrates extreme, obsessive curiosity for both the science and business
Matches our cultural phenotype: high ownership and agency, cares deeply, curious, ambitious, practical, unpretentious, collaborative, focus on shipping awesome products
Company
Axiom Bio
Axiom Bio helps scientists eliminate molecular toxicity by offering the most accurate and affordable predictive models.
Founded in 2024
11-50 employees
Funding
Current Stage
Early StageTotal Funding
unknown2025-07-01Seed
Recent News
2026-01-16
Company data provided by crunchbase