AI/ML Engineer

The Nuclear Company; Washington; On-site; USD 121,000 - 165,000; Full time

Benefits offered by this job

Competitive compensation packages

401k with company match

Medical, dental, vision plans

Generous vacation policy

Job summary

The Nuclear Company is seeking an AI Engineer to enhance the NOS software platform with machine learning and AI capabilities. This role involves integrating ML features into production, collaborating with engineers, and improving operations through data analysis.

With a focus on safety and efficiency, the successful candidate will have 5+ years of experience in ML, expertise in Python, and familiarity with large datasets. A competitive salary and comprehensive benefits package are offered.

Qualifications

• 5+ years of experience shipping ML or AI features into production.
• Strong Python and hands-on experience with modern ML tools.
• Experience with large datasets and building data pipelines.

Responsibilities

• Ship ML and AI features inside NOS.
• Build AI agents and workflows against NOS data.
• Own model quality end-to-end with various eval criteria.

Skills

Python programming

Machine Learning

Data Engineering

LLM tooling (PyTorch, Hugging Face)

Cloud experience (preferably AWS)

Education

Bachelor’s or Master’s in Computer Science or related field

Tools

MLflow

SageMaker

Vertex AI

Job description

The Nuclear Company is the fastest growing startup in the nuclear and energy space creating a never before seen fleet‑scale approach to building nuclear reactors. Through its design‑once, build‑many approach and coalition building across communities, regulators, and financial stakeholders, The Nuclear Company is committed to delivering safe and reliable electricity at the lowest cost, while catalyzing the nuclear industry toward rapid development in America and globally.

About the role

The Nuclear Company is hiring an AI Engineer to ship the machine learning and agent capabilities inside NOS, the software platform behind one of the largest nuclear buildouts in the United States. You’ll work in the Platform Integration & AI/Data squad and report to the Director of Platform Integrations.

This is a hands‑on engineering role. You’ll embed ML models and LLM‑driven workflows directly into the NOS applications our internal teams and partners use every day: site evaluation, red flag analysis, scheduling, lessons learned, and the stakeholder and project tools that run across active TNC projects. You’ll work close to the metal: training and fine‑tuning models, designing agent workflows against NOS data and tools, wiring up MCP interfaces, and standing up the evals and monitoring that keep AI features safe in production.

You’ll partner daily with platform engineers, product engineers, and the nuclear domain experts who use what you ship. The team is small, the pace is fast, and the work is visible, what you build will be demoed to utilities, regulators, and financial partners making 20‑to‑30‑year decisions about the future of American nuclear power.

Responsibilities

• Ship ML and AI features inside NOS. Train, fine‑tune, and deploy models and LLM‑based workflows into production, with the evals, monitoring, and safety controls that make them trustworthy.
• Build AI agents and agent workflows. Design agents that operate against NOS data and tools, including the MCP interfaces and tool‑use frameworks they depend on.
• Own model quality end‑to‑end. Define eval criteria, acceptance thresholds, and regression suites for AI features. Keep results reproducible and the bar high.
• Partner across the squad. Work directly with platform engineers and product engineers so models reach production instead of staying in notebooks.
• Contribute to the data ontology. Help shape how NOS data is structured so models and agents can use it reliably.
• Build predictive and anomaly‑detection models that support operations and engineering decisions, including time‑series analysis of sensor and operational data from active projects.

Required Experience

• Bachelor’s or Master’s in Computer Science, Machine Learning, Statistics, or a related technical field, or equivalent production experience.
• 5+ years of professional experience shipping ML or AI features into production. We care about what you’ve shipped, not just what you’ve studied.
• Strong Python and hands‑on experience with modern ML and LLM tooling (PyTorch, Hugging Face, or similar).
• Production experience with LLMs and agents: prompting, fine‑tuning, RAG, evals, tool use.
• Solid data engineering fundamentals, comfortable building data pipelines and working with large, messy datasets.
• Experience with at least one major cloud, AWS preferred.
• Strong written and verbal communication. You can explain models, tradeoffs, and failure modes to non‑specialists.

Preferred Experience

• Experience with Palantir Foundry and AIP. Given the central role Foundry plays in our platform, this is heavily weighted.
• Experience building agent systems with MCP or similar tool‑use frameworks.
• MLOps experience (MLflow, SageMaker, Vertex AI) and CI/CD for models.
• Experience in a regulated or safety‑critical industry: nuclear, aerospace, defense, energy, or medical.
• Background in time‑series, sensor, or operational data.
• Comfortable doing exploratory data analysis and translating findings into product decisions, in addition to shipping production models.

Benefits

• Competitive compensation packages
• 401k with company match
• Medical, dental, vision plans
• Generous vacation policy, plus holidays

Estimated Starting Salary Range

The estimated starting salary range for this role is $121,000 - $165,000 annually less applicable withholdings and deductions, paid on a bi‑weekly basis. The actual salary offered may vary based on relevant factors as determined in the Company’s discretion, which may include experience, qualifications, tenure, skill set, availability of qualified candidates, geographic location, certifications held, and other criteria deemed pertinent to the particular role.

EEO Statement

The Nuclear Company is an equal opportunity employer committed to fostering an environment of inclusion in the workplace. We provide equal employment opportunities to all qualified applicants and employees without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, age, disability, veteran status, or any other protected characteristic. We prohibit discrimination in all aspects of employment, including hiring, promotion, demotion, transfer, compensation, and termination.

Export Control

Certain positions at The Nuclear Company may involve access to information and technology subject to export controls under U.S. law. Compliance with these export controls may result in The Nuclear Company limiting its consideration of certain applicants.

Recruiting Fraud Alert

Your safety is our priority. We want to ensure your job search stays secure. Please note that the team at The Nuclear Company only communicates through official @thenuclearcompany.com email addresses. We will never ask for payments or sensitive financial information at any stage of our recruitment process. For your peace of mind, please verify all openings and submit your applications directly through our official careers page.