AI Research Scientist

About

Edison Scientific builds and commercializes AI agents for science. Scientific discovery moves too slowly, and autonomous AI agents are how we intend to fix that. We're assembling a team of top researchers and engineers across AI and biology to build an AI scientist.

Role

As an AI Research Scientist at Edison Scientific, you play a central role in developing the models and algorithms that accelerate scientific research. Your role will encompass both cutting-edge research and applied methods, bringing your work to bear on scientific challenges in laboratories and scientists around the world. 

This role is on-site at our San Francisco office in the Dogpatch neighborhood. Our office is a converted warehouse with high ceilings, open space, and a team excited about what we’re building.

Responsibilities

• 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.

Qualifications

• 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

$185,000 - $275,000  •  Offers equity

Why join us?

• 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 Carrot
• 401(k) company matching
• $300 health and wellness benefit
• Lunch is on us every day you're in the office, and dinner is on us when you're working late
• Regular team offsites and company events
• A fast-moving, mission-driven culture where smart people do their best work and actually enjoy doing it