Machine Learning Engineer

Why Nuclearn.ai

Nuclearn.ai builds AI-powered software for the nuclear and utility industries—tools that keep critical infrastructure reliable, efficient, and safe. Our software integrates AI-driven workflow, documentation, and research automation, and is already used at 60+ nuclear reactors across North America. You'll ship production code operators and engineers rely on every day.

We're growing quickly, expanding our team and our Phoenix HQ. The work is consequential: what you build helps real plants run safer and smarter.

Eligibility: U.S. citizenship or permanent residency (green card) is required due to DOE export compliance.

What You'll Do

* Collaborating closely with customers to understand their unique needs and tailoring AI solutions to meet specific industry challenges, particularly in the nuclear and utility sectors.

* Fine-tuning pre-trained language models for customer-specific classification, extraction, and prediction tasks.

* Designing, training, and validating custom ML pipelines to address domain-specific problems, ensuring high accuracy and performance in real-world applications.

* Implementing and optimizing ML models for deployment in production environments, with a focus on scalability and efficiency.

* Partnering with cross-functional teams, including development teams and domain experts, to ensure solutions align with customer workflows and objectives.

* Continuously improving models by leveraging customer feedback and incorporating new data.

* Driving innovation in the use of AI and ML within the nuclear and utility industries by experimenting with cutting-edge techniques and tools.

What Makes You a Great Fit

* Bachelor's or Master's degree in Computer Science, Machine Learning, or a related technical field

* 2+ years of experience implementing and deploying machine learning solutions, with at least 1 year of hands-on experience with language models

* Strong programming skills in Python and experience with PyTorch

* Demonstrated ability to translate technical capabilities into practical solutions

* Experience deploying models in production environments

Nice To Have (not Required)

* Experience deploying models in production environments

* Prior experience working in a startup environment

* Knowledge of the nuclear or utility industries

Compensation & Benefits

* Benefits: Unlimited PTO, health/dental/vision insurance

Work Model & Schedule

* Full-time, salaried

* Mon–Fri hybrid (Wed remote); expectation is ≥80% in-office (Phoenix HQ)

How We Hire (fast, Respectful, Practical)

* 20-min intro with the founder/hiring manager to trade context and assess mutual fit

* Practical work sample (60–90 min; a real task in our stack)

* Team meet + peer programming (system design + collaboration) We aim to move from first chat to decision quickly.