Research Scientist, Large-Scale Data Science and Learning

UT-Battelle; Oak Ridge (TN); On-site; USD 120,000 - 180,000; Full time

Benefits offered by this job

Medical and retirement plans

Flexible work hours

On-site amenities

Relocation assistance

Job summary

UT-Battelle's Oak Ridge National Laboratory invites applications for a Research Scientist in the AAIMS group to advance AI for science on leadership-class supercomputers, designing, training, and evaluating AI systems that plan, reason, and act to accelerate discovery across materials, chemistry, climate, fusion, biology, and more.

You will collaborate with senior researchers, publish results, and contribute to open-source software, leveraging Frontier exascale resources and world-class

Qualifications

• Ph.D. in Computer Science, Computer Engineering, or a closely related field.
• Demonstrated research in one or more areas of HPC or AI (e.g., large-scale training, scientific reasoning, reinforcement learning, or distributed systems).
• Strong programming skills (Python, C/C++, or equivalent) and experience with ML frameworks (e.g., PyTorch).

Responsibilities

• Conduct research in AI/ML at scale, working with cutting-edge HPC resources.
• Collaborate with senior researchers and domain scientists on AI methods and scientific applications.
• Contribute to peer-reviewed publications, technical reports, and proposals.
• Engage in collaborative software development and open-source contributions.
• Present research outcomes at conferences, workshops, and internal seminars.
• Contribute to a supportive, inclusive, and collaborative team culture.

Skills

Python

C/C++

PyTorch

HPC/Distributed training

Reinforcement Learning

Education

Ph.D. in Computer Science or related field

Tools

DeepSpeed

Megatron-LM

Ray

Distributed training frameworks

Job description

Overview

The Analytics and AI Methods at Scale (AAIMS) group in the National Center for Computational Science (NCCS) is hiring a Research Scientist to advance the frontier of AI for science, including scientific reasoning, federated & collaborative learning, and reinforcement learning (RL) for self-improving models on leadership-class supercomputers.

Requisition Id 16411

Overview

The Analytics and AI Methods at Scale (AAIMS) group in the National Center for Computational Science (NCCS) is hiring a Research Scientist to advance the frontier of AI for science, including scientific reasoning, federated & collaborative learning, and reinforcement learning (RL) for self-improving models on leadership-class supercomputers. You’ll help design, train, and evaluate AI systems that plan, reason, and take actions to accelerate discovery across domains (materials, chemistry, climate, fusion, biology, and more). NCCS operates the Frontier exascale supercomputer and world-class data facilities. This role sits at the intersection of AI at scale and HPC, giving you access to unmatched resources to prototype new ideas, run experiments, and translate methods into scientific impact.

Examples Of Focus Areas

• Agentic AI for Science: Autonomous and tool-using agents for experiment design, simulation steering, data collection, and lab/compute orchestration; planning and memory; multi-agent collaboration.
• Scientific Reasoning: Program/path-of-thought, tool-augmented and retrieval-augmented reasoning; uncertainty quantification and calibrated decisions.
• RL & Self-Improving Models: RLHF/RLAIF, online RL, self-play, open-ended discovery, reward modeling, curriculum/active learning, data selection, iterative post-training, safety alignment and guardrails.
• Foundation Models for Science @ Scale: Pretraining, instruction tuning, continued pretraining, Mixture-of-Experts; distributed training/inference (FSDP, DeepSpeed, Megatron-LM, tensor/sequence parallelism); scalable evaluation pipelines for reasoning and agents.
• Federated & Collaborative Learning: Cross-silo training across institutions and facilities; privacy-preserving learning (secure aggregation, differential privacy, MPC/HE); personalization under heterogeneity; governance-aware data/model sharing; collaborative evaluation.

Major Duties And Responsibilities

• Conduct research in AI/ML at scale, working with cutting-edge HPC resources.
• Collaborate with senior researchers and domain scientists on AI methods and scientific applications.
• Contribute to peer-reviewed publications, technical reports, and proposals.
• Engage in collaborative software development and open-source contributions.
• Present research outcomes at conferences, workshops, and internal seminars.
• Contribute to a supportive, inclusive, and collaborative team culture.

Basic Qualifications

• Ph.D. in Computer Science, Computer Engineering, or a field closely related to the job duties of this position.
• Demonstrated research in one or more areas of HPC or AI (e.g., large-scale training, scientific reasoning, reinforcement learning, or distributed systems).
• Strong programming skills (Python, C/C++, or equivalent) and experience with ML frameworks (e.g., PyTorch).

Preferred Qualifications

• Experience with large-scale experiments on HPC or cloud platforms.
• Strong publication record commensurate with career stage.
• Familiarity with distributed training frameworks (e.g., DeepSpeed, Megatron-LM, Ray).
• Demonstrated ability to work collaboratively in multidisciplinary research teams.
• Interest in developing open-source tools and contributing to community efforts.

Special Requirements

• Please submit two letters of reference when applying to this position.

About ORNL

As a U.S. Department of Energy (DOE) Office of Science national laboratory, ORNL has an impressive 80-year legacy of addressing the nation’s most pressing challenges. Our team is made up of over 7,000 dedicated and innovative individuals! Our goal is to create an environment where a variety of perspectives and backgrounds are valued, ensuring ORNL is known as a top choice for employment. These principles are essential for supporting our broader mission to drive scientific breakthroughs and translate them into solutions for energy, environmental, and security challenges facing the nation.

ORNL offers competitive pay and benefits programs to attract and retain individuals who demonstrate exceptional work behaviors. The laboratory provides a range of employee benefits, including medical and retirement plans and flexible work hours, to support the well-being of you and your family. Employee amenities such as on-site fitness, banking, and cafeteria facilities are also available for added convenience.

• Prescription Drug Plan
• Dental Plan
• Vision Plan
• 401(k) Retirement Plan
• Contributory Pension Plan
• Life Insurance
• Disability Benefits
• Generous Vacation and Holidays
• Parental Leave
• Legal Insurance with Identity Theft Protection
• Employee Assistance Plan
• Flexible Spending Accounts
• Health Savings Accounts
• Wellness Programs
• Educational Assistance
• Relocation Assistance
• Employee Discounts

This position will remain open for a minimum of 5 days after which it will close when a qualified candidate is identified and/or hired.

ORNL is an equal opportunity employer.

All qualified applicants, including individuals with disabilities and protected veterans, are encouraged to apply.

UT-Battelle is an E-Verify employer.