Machine Learning Scientist, AI Explainability

At SES AI, base pay is one part of our total compensation package and is determined within a range. The base pay range for this role is between $150,000 and $250,000. SES AI considers several factors when extending an offer, including but not limited to, the role, function and associated responsibilities, a candidate’s work experience, location, education/training, and skills.

What We Offer

• Company paid Health and Dental insurance (ability to add dependents)
• Global travel insurance for employees traveling while on business
• Company sponsored retirement plan with 100% vesting and up to 5% match.
• Life and AD&D Insurance
• Employee Assistance Program
• Six Paid Holidays, and one floating holiday per a quarter equivalent to 4 per calendar year
• 10 accrued vacation days per calendar year that increases with tenure.
• Bonus + Equity, based on position and eligibility requirements

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

About SES AI:
SES AI Corp. (NYSE: SES) is powering the future of global electric transportation on land and in the air with the world’s most advanced Li-Metal batteries. SES AI is the first battery company in the world to accelerate its pace of innovation by utilizing superintelligent AI across the spectrum of its business, from research and development; materials sourcing; cell design; engineering and manufacturing; to battery health and safety monitoring. Founded in 2012, SES AI is an Li-Metal battery developer and manufacturer headquartered in Boston and with operations in Singapore, Shanghai, and Seoul.

Learn more at SES.AI

Position Scope

We are a leading company specializing in the development of advanced lithium-metal batteries, on the brink of leveraging Large Language Models (LLMs) and extensive datasets to discover new battery materials. Our mission is to push the boundaries of what's possible in battery technology through innovative AI applications. To this end, the SES AI Prometheus team (AI Research) is seeking a highly skilled Machine Learning Scientist with a focus on AI explainability. This role will delve into the intricacies of large language model behavior, akin to studying the "brain" of LLMs or multimodal LLMs, utilizing causal graphs/representation, mechanistic interpretation and representation engineering tools etc. The objective is to extract the reasoning and planning capabilities of the LLMs and guide the machine towards creating innovative battery formulas.

You will be required to collaborate with strong academic labs, engaging in machine learning research aimed at addressing our battery design challenges and enhancing our systems' ability to understand and interpret data-driven science efficiently. Your contributions will be instrumental in enhancing our ability to analyze experimental data and intuitively achieve groundbreaking advancements in battery technology.

This is a remote position.

Responsibilities

• Conduct groundbreaking research on the explainability of large language models and their reasoning processes, drawing parallels with neuroscience to understand the "thought processes" of AI systems.
• Investigate the mechanisms through which LLMs approach problem-solving, planning, and solution generation, particularly in the context of basic battery design questions.
• Apply advanced techniques such as causal graphs and neuroscience-inspired AI methodologies to dissect and enhance the reasoning capabilities of LLMs, with the aim of improving accuracy and reducing instances of hallucination.
• Collaborate closely with a multidisciplinary team to integrate findings into practical AI solutions that contribute to the discovery of new battery materials and the advancement of lithium-metal battery technology.
• Contribute to academic and industry discussions by publishing research findings in top-tier journals and presenting at conferences.

Qualifications

• MS or PhD in Computer Science, Statistics, Computational Neuroscience, Cognitive Science or a related field, or equivalent practical experience.
• Strong foundational knowledge and practical experience in Machine Learning, Deep Learning, and Large Language Models.
• Proficiency in utilizing causal graphs for AI research and application.
• Demonstrated experience in AI explainability, with a focus on mechanistic interpretation and representation engineering.
• A solid track record of innovative research, preferably with published work in relevant areas.
• Proficient in programming languages relevant to machine learning, with a strong preference for Python.
• Experience with deep learning frameworks such as PyTorch or Tensorflow etc.
• Excellent problem-solving abilities and a passion for tackling complex technical challenges.
• The ability to communicate complex concepts clearly and effectively to both technical and non-technical team members.

Preferred Qualifications

• Experience with AI applications in material science or battery technology.
• Familiarity with the latest trends and methodologies in AI research, including neuroscience-inspired models and causal reasoning.