AIML - Machine Learning Research Scientist, Data and ML Innovation

Join a research team focused on understanding and expanding the capabilities of foundation models in scientific domains. This role involves conducting novel research, publishing findings, and applying them to product development. You will design experiments, develop methods, analyze results, and collaborate across teams to solve practical problems using insights from foundation models.

Must Have

• Perform fundamental research on foundation models for scientific domains.
• Define projects, develop methods, and design experiments.
• Run, analyze, and interpret experimental results.
• Publish research papers in top-tier venues.
• Apply research findings to solve practical problems.
• Collaborate with internal teams.
• Hold a PhD in computer science, statistics, physics, chemistry, electrical engineering, or operations research.
• Have 3 publications in top-tier ML, statistics, or NLP venues.
• Possess deep knowledge of foundation models and experience training them.
• Experience designing experiments to understand foundation model workings.
• Knowledge of Bayesian statistical methods for scientific inference.

Good to Have

• Experience with large language models (training and inference with tools like vllm).
• Proficient implementing ML models and experiments in Python and Pytorch/Jax.
• Familiarity with interpretability methods like activation patching/causal tracing.
• Demonstrate the ability to refine ambiguous research ideas to construct a coherent and logically sound story.

Perks & Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• Range of discounted products and free services
• Reimbursement for certain educational expenses (including tuition)
• Opportunity to become a shareholder through employee stock programs
• Ability to purchase stock at a discount through Employee Stock Purchase Plan
• Eligibility for discretionary bonuses or commission payments
• Eligibility for relocation

Would you like to join a team curious about understanding how foundation models work and to expand their capabilities in scientific domains? We perform and publish novel research and apply our findings to drive product directions. If you like designing clever approaches to understanding complex phenomena and using that knowledge to solve practical problems then this is the position for you.

In this research scientist role you will join a small team of researchers performing fundamental research investigating foundation models for scientific domains. You will be involved in all aspects of performing research including project definition, method development, and experimental design. You will also run your own experiments, then analyze and interpret the results. You will take an active role in writing papers for publication and also using our findings to solve applied problems. You will also have the opportunity to collaborate with partner teams across the company.

• Design robust experiments and methods to understand what foundation models do under the hood.
• Implement your methods and designs in experiment pipelines.
• Analyze and interpret your experimental results.
• Communicate results to teams across the company.
• Write papers for publication in top-tier conferences/journals.

Key Qualifications

• PhD in computer science, statistics, physics, chemistry, electrical engineering, or operations research. Other hard sciences may also be considered.
• 3 publications in top-tier machine learning, statistics, or natural language processing venues.
• Deep knowledge of foundation models and experience training them and applying them to real, complex datasets as demonstrated through publications or code repositories.
• Experience designing experiments to understand how foundation models work.
• Knowledge of Bayesian statistical methods and how they are used for scientific inference.

Additional Requirements

• Experience with large language models and – both training them and using tools like vllm for inference.
• Proficient implementing ML models and experiments in Python and Pytorch/Jax.
• Familiarity with interpretability methods like activation patching/causal tracing.
• Demonstrate the ability to refine ambiguous research ideas to construct a coherent and logically sound story.

At the company, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $139,500 and $258,100, and your base pay will depend on your skills, qualifications, experience, and location.

Company employees also have the opportunity to become a shareholder through participation in discretionary employee stock programs. Company employees are eligible for discretionary restricted stock unit awards, and can purchase stock at a discount if voluntarily participating in the Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at the company, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Company Benefits.

Note: Company benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

The company is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant

.

The company accepts applications to this posting on an ongoing basis.