ML Scientist
Pioneers innovative ML techniques and builds foundation models for clinical information extraction and synthesis from medical records. Requires PhD in CS/math with NLP/ML focus, high-impact publications, and experience with large-scale model training using PyTorch/JAX.
About the job
Responsibilities
- Design and implement state-of-the-art machine learning techniques to advance Layer Health's research agenda (in areas such as information extraction, multimodal reasoning, and summarization).
- Propose new agentic methods that tackle fundamental NLP and ML challenges such as modeling over multiple documents, long contexts, multiple modalities, and with limited or noisy labels.
- Build foundation models to power the future of clinical information extraction & synthesis, from training through inference.
- Stay up-to-date and actively engage with cutting-edge research in NLP, generative AI, and clinical machine learning.
- Collaborate with the broader engineering team to ship performant products that meet user needs.
- Cultivate and foster a robust and thoughtful R&D culture that drives the company forward.
Requirements
- Exceptional methodological research background and experience, including a PhD in computer science/applied mathematics or equivalent research experience, specializing in natural language processing and machine learning.
- High-impact, early-author publications at top peer-reviewed ML journals/conferences.
- Demonstrated record of delivering real-world impact from start to finish – with the ability to design, develop, and ship innovations.
- Strong programming skills and fluency with modern machine learning/LLM stacks (PyTorch, Jax).
- Past experience in training/inference of foundation models (billions of parameters, distributed training, familiarity with state-of-the-art techniques).
- A strong communicator who thrives in a customer-focused, fast-paced environment.
- An excited and adaptable team player who wants to disrupt the healthcare industry with AI/ML.
Nice-to-Haves
- Past experience in healthcare or life sciences.
Compensation
- Expected compensation range: $200,000-250,000, in addition to stock options. Compensation dependent on experience, fit, and location.
Skills
PyTorch, JAX, Natural Language Processing, Machine Learning, LLMs, Foundation Models, Distributed Training, Information Extraction, Multimodal Reasoning, Generative AI
Similar jobs
AI Research jobs
Mercor
San Francisco, CA
Leads the design, measurement, publication, and adoption of APEX benchmarks evaluating frontier models on economically valuable professional work. The role requires rigorous research judgment, strong coding and statistical skills, and excellent communication across technical, commercial, and research audiences.
Tessera Labs
San Jose, CA
Research Scientist defining and executing research on reliable long-horizon agents in enterprise environments. The role focuses on post-training and reinforcement learning, agent memory, evaluation, verification, and structured representations, combining hands-on experimentation with product delivery and publication.
OpenAI
San Francisco, CA
Conduct rigorous people research and applied data science to evaluate talent programs, organizational health, and employee experiences. The role requires advanced expertise in research design, experimentation, measurement, causal inference, statistical modeling, and responsible handling of sensitive employee data.
Earnin
Mountain View, CA
Build agent-driven chatbots and generative AI workflows for financial-wellness products, owning features from design through impact measurement. The role requires at least three years of software engineering experience, strong system design, maintainable coding practices, and a bachelor’s degree or equivalent experience.
Scale AI
San Francisco, CAResearch Scientist focused on evaluating frontier language and multimodal models, diagnosing failure modes, and building rigorous benchmarks. The role requires advanced training in AI or a related field, post-training expertise, and published machine learning research.