Senior AI Research Scientist: Scientific ML & LLMs
$210,000 - $350,000
Apply now (opens in a new tab)Senior AI Research Scientist: Scientific ML & LLMs
L5
$210,000 - $350,000
Industries
Research and Development in the Physical, Engineering, and Life SciencesOther Scientific and Technical Consulting ServicesAll Other Professional, Scientific, and Technical Services- Interpret challenges in building AI agents for science as well-formulated statistical problems
- Conduct research in post-training of LLMs on challenging scientific reasoning tasks, with the goal of exceeding expert human capacity
- Conduct research in optimal sampling techniques, considering aspects like efficient exploration and quality-diversity
- Own research projects end-to-end, from working with scientists to scope the problem all the way to working with engineers to deploy the outcomes
- Develop and extend our experimentation platform for internal tools and projects.
- Collaborate closely with a multidisciplinary team of AI researchers, engineers, biologists, and chemists, fostering an environment of innovation and discovery.
- 5-10+ years of strong track record of ML research on real-world problems in (but not limited to) reinforcement learning, representation learning, and Bayesian methods
- Fluency in PyTorch, Jax, or equivalent framework
- Experience with large-scale training and experimentation technologies (e.g. Ray, verl, Megatron-LM, etc.)Experience working with data and ML systems: EDA, feature engineering, model development, model deployment, and validation in production systems.
- Familiarity with modeling in PyTorch, Jax or equivalent framework.
- Demonstrated experience with experimentation in academic or industry settings.
- Strong programming expertise with the capability to adapt to various technical challenges in the data, ML, and LLM software stack.
- PhD in Machine Learning, Computer Science, or other quantitative field
- Familiarity with leveraging and managing distributed systems
- Salary$210,000 - $350,000 • Offers equity
- Competitive salary and equity
- Full healthcare coverage; we pay 100% of premiums for you and your dependents
- Support for growing families, including a yearly new parent stipend and fertility coverage through CarrotMental health support through Rula, our in-network therapist and psychiatrist network with fast availability12 weeks of paid parental leave for maternity, paternity, and adoption
- Pet care support with a yearly employer-funded stipend for your animal companions
- Commuter benefits so you can pay for transit and parking with pre-tax dollars 401(k) company matching$300 health and wellness benefit quarterly
- Lunch is on us every day you're in the office, and dinner is on us when you're working late
- Regular team off-sites and company events
Level
SeniorL5
Salary
$210,000 - $350,000
Location
Millbrae, CA
Occupation
Computer and Information Research Scientists
Industry
Research and Development in the Physical, Engineering, and Life Sciences
Posted
today
To get sharper similar jobs, create your profile using the link below.