[Remote] Software Engineer (Applied AI/ML)

Note: The job is a remote job and is open to candidates in USA. Oklo is pioneering advanced fission power technology to provide clean, reliable, and affordable energy. The Software Engineer will develop and deploy AI/ML solutions for nuclear system design and analysis, create LLM-based workflows and digital twin surrogate models, and lead software engineering initiatives while presenting progress to senior leadership.

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

• * 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
• * 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
• * 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
• A nuclear background is not required, but a healthy curiosity and eagerness to learn are essential in this role
• Oklo requires remote employees to travel to headquarters (Santa Clara, CA) twice a quarter annually, based on business or team needs, including attendance at team meetings, off-sites, and other company events or gatherings
• For the first two weeks of onboarding, employees are required to be in person at headquarters in Santa Clara, CA
• This position may involve access to information subject to U.S. export control laws. Only applicants who meet the definition of a U.S. person under applicable laws may be eligible
• * 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.)

Benefits

• Flexible time off
• Equity
• Bonuses
• 401(k)
• Health insurance (with employer contribution)
• HSA
• FSA
• Flexible work hours
• Wellness credits

Company Overview