Software Engineer (Applied AI/ML)

Oklo; United States; On-site; USD 170,000 - 210,000; Full time

Job summary

Oklo is hiring a Software Engineer to develop ML-enabled workflows and production models for nuclear system design and analysis. You will work with engineers to deploy surrogate models and real-time digital twins while growing the software team.

The role emphasizes security, fast iteration, and clear communication with senior leadership, aiming to advance clean energy through advanced fission technologies.

Qualifications

• Bachelor's degree in Computer Science, Electrical Engineering, Math, Physics, or equivalent software engineering experience.
• Managing and/or integrating AI into large, corporate data ontologies.
• Leading company-wide initiatives (from project inception to delivery and maintenance).
• Physics-informed machine learning (surrogates, symbolic encodings, digital twins).
• Python fluency.
• CI/CD pipelining.
• Git proficiency.
• Growing small software organizations.
• Working with large data collection pipelines.
• Optimization, statistics and design of experiments.

Responsibilities

• Invent and implement practical LLM-based workflows to accelerate Oklo's business and engineering processes.
• Ensure rigorous security and information control.
• Accelerate both engineering and non-technical workflows.
• Design, implement, deploy and maintain production machine learning models for the design and analysis of nuclear systems.
• Develop surrogate models for use in real-time digital twins.
• Grow and manage a team of software engineers to support your AI initiatives.
• Present roadmaps, product demos, and progress updates to senior leadership.
• Ensure their departments' needs are met by your initiatives.
• Solicit user feedback from individual contributors and department leadership.

Skills

Python fluency

AI/ML integration

CI/CD

Git proficiency

Data pipelines

Optimization & DOE

Team leadership

Nuclear domain curiosity

Physics-informed ML

Education

Bachelor's degree in CS/EE/Math/Physics

Tools

Rust

Typescript/Javascript (Svelte)

GitLab CI/CD

MCNP/SERPENT/Open-MC

Job description

Thanks for your interest in Oklo! We are searching for a Software Engineer to join our team. Join us in pioneering the next generation of nuclear reactors! You'll leverage your software skills alongside nuclear engineers to model, simulate, design, and deploy advanced fission power technology. You will work at the forefront of the nuclear industry, developing novel techniques to reach new levels of safety, efficiency, and resiliency. Come be a part of powering the future with advanced fission power plants to provide clean, reliable, affordable energy.

Position Description

Oklo is a fast-moving startup where you will be responsible for tackling a wide range of challenges in deploying nuclear systems. A nuclear background is not required, but a healthy curiosity and eagerness to learn are essential in this role.

Specific responsibilities may include:

• Invent and implement practical LLM-based workflows to accelerate Oklo's business and engineering processes.
• Ensure rigorous security and information control.
• Accelerate both engineering and non-technical workflows.
• Design, implement, deploy and maintain production machine learning models for the design and analysis of nuclear systems.
• Develop surrogate models for use in real-time digital twins.
• Grow and manage a team of software engineers to support your AI initiatives.
• Present roadmaps, product demos, and progress updates to senior leadership.
• Ensure their departments' needs are met by your initiatives.
• Solicit user feedback from individual contributors and department leadership.

Minimum Qualifications:

• Bachelor's degree in Computer Science, Electrical Engineering, Math, Physics, or equivalent software engineering experience.
• Managing and/or integrating AI into large, corporate data ontologies.
• Leading company-wide initiatives (from project inception to delivery and maintenance).
• Physics-informed machine learning (surrogates, symbolic encodings, digital twins).
• Python fluency.
• CI/CD pipelining.
• Git proficiency.
• Growing small software organizations.
• Working with large data collection pipelines.
• Optimization, statistics and design of experiments.

Bonus Qualifications:

• Rust development
• Typescript/Javascript (especially Svelte) development
• GitLab CI/CD
• Growing and managing teams of software engineers.
• Navigating sensitive, controlled, or externally regulated data.
• Knowledge of and excitement for a nuclear-powered future.
• Navigating quality assurance and compliance requirements with AI.
• Nuclear engineering concepts (reactors, power generation, fuel performance, fuel recycling, etc.)

Competencies

We are looking for a Software Engineer that is:

• Passionate about clean energy
• Passionate about making advanced fission a reality
• Willing and able to learn quickly
• Knowledgeable about fast reactor design
• Highly skilled with Monte Carlo neutronics tools such as MCNP, SERPENT, and Open-MC
• Skilled with deterministic fast reactor tools, or wiling and eager to learn
• Eager to work on a dynamic team, receive constructive feedback, and grow with us
• Willing to think differently and do things in new ways
• Comfortable in a fast-paced, highly iterative startup environment
• Excited to think creatively, critically, and reflectively about the problems they are solving, while not leaning only on what has been done before
• Willing to propose novel and creative solutions to technical problems
• An excellent writer who can write in modern active voice, so make your cover letter compelling and write it well!
• Able to communicate technical content and results verbally

Who you are:

A startup person : You aren't driven by titles or hierarchy, and prefer efficiency to excess process. You don't need or expect to have a lot of guidance but you enjoy working in a fast-paced team. If you prefer the culture and feel of a large organization, that is great, but you likely won't enjoy working with us! There is plenty of important work and plenty of good opportunities with organizations like that.

Motivated : You are self-motivated. You bring an enthusiasm to the team, and imbue a sense of passion that goes beyond clocking in and clocking out. This isn't about a fake or arbitrary "pieces of flair" mentality or lack of work-life balance! It is about being a part of the vision and feeling a part of reaching team goals.

A team-player : Oklo genuinely is a team. We aren't about taking credit for ourselves, and we aren't about pushing blame to others. We do incredible things because we work as a team.

An excellent communicator : We need a person who is not only technically competent but also a clear and upbeat communicator.

Creative : Being creative means that when things fall outside clear scopes or processes or problems arise without clear solutions, you are able to identify it as well as invent ways to solve a problem or fill a need without micromanagement. The successful person in this job will not only be creative, but also enjoy being creative and solving