Skip to main content

Oklo

Software Engineer (Applied AI/ML)

Santa Clara, CA or Remote (Remote)$200k–$250kmidAdded today

About this role

Oklo seeks a Software Engineer to develop AI/ML solutions for advanced nuclear reactor design and deployment. You'll build LLM-based workflows, production ML models, and digital twin surrogates while leading a growing software team to support the company's engineering and business operations.

What you'll do

  • Create and deploy LLM-based workflows for nuclear engineering and business processes with security controls
  • Design, implement, and maintain production ML models for nuclear system design and analysis
  • Develop surrogate models for real-time digital twin applications
  • Build and manage a team of software engineers supporting AI initiatives
  • Lead cross-functional projects from inception through delivery and ongoing maintenance
  • Present technical roadmaps and product updates to senior leadership while gathering stakeholder feedback

What they're looking for

  • Python development
  • Physics-informed machine learning and surrogate modeling
  • Large language model integration and prompt engineering
  • CI/CD pipeline design and implementation
  • Git version control
  • Data collection and pipeline management
  • Optimization, statistics, and experimental design
  • Team leadership and software organization growth
Apply with Autofill

Opens the application — the Jobs AI extension fills it for you. Set up autofill

Opens the official application on the employer’s site. No login required.

Oklo

Oklo develops advanced fission reactor technologies aimed at commercializing clean energy solutions. The company is hiring regulatory engineers, thermal hydraulic test engineers, technical product engineers, systems engineers, and software lifecycle specialists to support reactor design, licensing, testing, and engineering operations.

View all jobs at Oklo

Likely interview questions

  • Describe your experience integrating AI into large corporate data systems and what challenges you encountered managing data governance.
  • Tell us about a physics-informed ML project you've led—how did you validate surrogate model accuracy against domain physics?