AI/ML Software Engineer

Location: Remote, Continental US time zones

About Atomic Canyon

At Atomic Canyon, our vision is to unlock endless energy. We are pioneering the development of safe, efficient, open-source AI models tailored for nuclear data and content. In collaboration with Oak Ridge National Laboratory (ORNL) and the world’s fastest supercomputer, Frontier, our cutting-edge AI technology processes nuclear terminology with unmatched precision, positioning us to solve complex challenges in nuclear energy.

Our mission is to harness AI to power nuclear progress, starting with our flagship product, Neutron—a generative AI search solution for nuclear energy plants. By simplifying access to vast documentation, enhancing data accessibility, and streamlining workflows, we’re using advanced AI to address critical climate and energy challenges—and revitalize the nuclear energy sector.

Atomic Canyon was founded by Trey Lauderdale, who previously led Voalte (acquired for $180M), and Kristian Kielhofner, formerly of Star 2 Star Communications (acquired for $437M).

About the role

We're seeking an AI/ML Software Engineer to join our innovative team and help build advanced AI models for the nuclear sector. You'll work on a range of systems, including document understanding, RAG, and domain-adapted language models. Your work will directly contribute to making nuclear energy information more accessible and promoting industry knowledge-sharing.

What you will do and achieve

  • Develop and implement AI models to understand complex nuclear content and concepts

  • Design and optimize AI models for efficient data navigation and document indexing

  • Collaborate with ORNL scientists to leverage supercomputing resources for model training

  • Implement safeguards to prevent AI hallucinations and biases during search and data analysis

  • Create robust, scalable, and maintainable code for our AI platform

  • Contribute to the development of open-source AI tools for the nuclear sector

  • Work closely with nuclear experts to ensure AI models accurately interpret industry-specific content

Who you are

  • 3+ years of experience in AI/ML software engineering

  • Strong background in natural language processing, machine learning, and large language and information retrieval models

  • Experience with high-performance distributed computing environments is a plus

  • Familiarity with or interest in the nuclear energy sector

Key knowledge and skills:

  • Proficiency in Python and experience with PyTorch

  • Strong understanding of AI models, particularly in document understanding and information retrieval

  • Hands-on experience implementing and fine-tuning Retrieval-Augmented Generation (RAG) systems for domain-specific applications

  • Experience with the HuggingFace ecosystem

  • Experience with large-scale data processing and distributed computing

  • Familiarity with open-source development practices and version control systems

  • Excellent problem-solving skills and ability to work in a fast-paced, innovative environment

  • Strong communication skills to collaborate with cross-functional teams and external partners

  • Commitment to safety and security in dealing with nuclear-related information

Benefits

  • Competitive salary of $150,000-$225,000 annually

  • Equity in the company

  • Fully remote work environment

  • Comprehensive health, dental, and vision insurance

  • Empowering, flexible work environment that supports both professional goals and individual passions

  • Professional development opportunities, including conference attendance and training programs

  • Opportunity to work with world-class supercomputing facilities

  • Chance to make a significant impact on the future of clean energy

  • Quarterly travel to rotating US destinations for in-person company meetings

Atomic Canyon is committed to fostering a diverse and inclusive workplace. We do not discriminate based on race, color, religion, gender, sexual orientation, national origin, age, disability, or any other protected characteristic. We encourage applications from all qualified individuals who share our passion for innovation in nuclear energy and AI.