AI Research Scientist at Edison Scientific

San Francisco, California, United States -

Full Time

Start Date

Immediate

Expiry Date

05 Jun, 26

Salary

275000.0

Posted On

07 Mar, 26

Experience

5 year(s) or above

Remote Job

Yes

Telecommute

Yes

Sponsor Visa

Skills

Reinforcement Learning, Representation Learning, Bayesian Methods, PyTorch, Jax, Ray, Verl, Megatron-LM, EDA, Feature Engineering, Model Development, Model Deployment, Validation, LLM Software Stack, Distributed Systems

Industry

Software Development

Description

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. Bonus points for: 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

Responsibilities

The AI Research Scientist will develop models and algorithms to accelerate scientific research by interpreting scientific challenges as statistical problems and conducting research on LLM post-training and optimal sampling techniques. Responsibilities also include owning research projects end-to-end, developing the experimentation platform, and collaborating with multidisciplinary teams.