Staff Scientist - Post-Training and Reinforcement Learning for AI for Science

Argonne National Laboratory; United States; Hybrid; USD 94,000 - 147,000; Full time

Benefits offered by this job

Comprehensive benefits

Job summary

Argonne National Laboratory seeks a Staff Scientist in Post-Training and Reinforcement Learning for AI for Science to advance foundation models and learning systems for scientific discovery at leadership-class supercomputers.

You will develop, scale, and evaluate post-training methods including reinforcement learning and alignment techniques for scientific AI models and workflows, collaborating across AI, simulation, and domain science while contributing to high-impact publications and

Qualifications

• Bachelor's degree or higher in CS or related field.
• Advanced knowledge in machine learning, reinforcement learning, large-scale model training, post-training, optimization, data mining, or statistics.
• Strong background in mathematical optimization, linear algebra, or numerical methods.
• Programming experience in Python, C, or C++.
• Experience with ML frameworks such as PyTorch or JAX.
• Experience with large-scale training and post-training workflows.
• Ability to work in interdisciplinary teams.

Responsibilities

• Conduct research and development aligned with Argonne's mission in computation, AI, and scientific discovery.
• Develop, scale, and optimize post-training methods for scientific foundation models, including RL, preference-based optimization, fine-tuning, alignment.
• Advance techniques to improve performance, controllability, and reliability of AI models for science.
• Design and evaluate methods for applying RL and post-training pipelines to large-scale environments.
• Develop and optimize workflows for training on leadership-class supercomputers and AI-oriented architectures.
• Collaborate with computational scientists and domain researchers on challenging problems.
• Address algorithmic, systems, and data challenges with large-scale training and post-training.

Skills

Python

C/C++

Machine Learning / Reinforcement

Optimization

Distributed training

Communication skills

Interdisciplinary collaboration

Education

Bachelor's degree

Master's degree

PhD

Related field degree

Tools

PyTorch

JAX

TensorFlow

Linux

Job description

The Argonne Leadership Computing Facility (ALCF) is seeking a Staff Scientist in Post-Training and Reinforcement Learning for AI for Science to help advance the next generation of foundation models and learning systems for scientific discovery.

This is an opportunity to work at the frontier of AI for science and the Department of Energy Genesis mission, where large-scale machine learning, scientific data, simulation, and leadership-class supercomputers come together to enable new modes of discovery across physics, materials science, chemistry, biology, climate, energy, and related fields. We are looking for a creative and collaborative scientist who is excited to develop, scale, and evaluate post-training methods, including reinforcement learning, preference optimization, adaptation, and alignment techniques, for scientific AI models and workflows.

The successful candidate will conduct research on methods that improve the usefulness, reliability, and scientific performance of large-scale AI models after pretraining, while also advancing the systems and software needed to run these methods efficiently on cutting-edge supercomputers and emerging AI platforms. This role offers the opportunity to contribute both fundamental advances in machine learning and high-impact scientific applications while working in a multidisciplinary environment with experts in AI, simulation, computer science, applied mathematics, and domain science.

You will join the AI group - a highly collaborative, multidisciplinary environment and work alongside experts in AI, simulation, computer science, applied mathematics, and domain science. This role offers the chance to contribute both foundational advances and real-world scientific outcomes, with opportunities to publish in leading journals and conferences, engage with national and international collaborators, and influence AI and HPC for scientific research.

In this role you will:

• Conduct research and development aligned with Argonne's strategic mission in computation, AI, and scientific discovery.
• Develop, scale, and optimize post-training methods for scientific foundation models, including reinforcement learning, preference-based optimization, fine‑tuning, alignment, and related approaches.
• Advance techniques that improve the performance, controllability, reliability, and scientific utility of AI models for science applications.
• Design and evaluate methods for applying reinforcement learning and post‑training pipelines to large‑scale scientific and data‑intensive environments.
• Develop and optimize workflows for training and post‑training on leadership‑class supercomputers and emerging AI‑oriented architectures.
• Partner with computational scientists, applied mathematicians, and domain researchers to apply foundation models and adaptive learning systems to challenging scientific problems with high impact.
• Address algorithmic, systems, and data challenges associated with large‑scale training and post‑training, including performance, scalability, robustness, and usability.
• Conduct original research in computational science and AI at scale, and communicate findings through publications, conference presentations, software, reports, and other research outputs.
• Work closely with colleagues across national laboratories, universities, industry, and supercomputing centers on current and future systems for the AI for science mission.
• Contribute to a team culture that values scientific excellence, collaboration, innovation, and inclusive professional growth.

"This position qualifies as Hybrid Remote Work - Mostly Onsite": which applies to employees regularly scheduled for some onsite and some remote days, with employees typically working up to 40% of their time remotely.

Position Requirements

Required Qualifications:

• RD2: Bachelor's degree and 5+ years of experience, or a Masters and 3+ years of experience, or a PhD, or equivalent
• Education in computer science, applied mathematics, statistics, computational science, or a related field
• Demonstrated advanced knowledge in one or more of the following areas: machine learning, reinforcement learning, large-scale model training, post-training, optimization, data mining, or statistics
• Strong background in mathematical optimization, linear algebra, or numerical methods
• Advanced knowledge of and significant programming experience in one or more languages such as Python, C, or C++
• Significant experience with machine learning frameworks such as PyTorch or JAX
• Experience with large-scale training, distributed learning systems, or post‑training workflows
• Experience with software development practices and techniques for computational science and machine learning systems
• Ability to work effectively in interdisciplinary teams involving mathematicians, computer scientists, and application scientists
• Effective written and verbal communication skills
• Ability to model Argonne's core values of impact, safety, respect, integrity, and teamwork

Preferred Qualifications:

• Experience with reinforcement learning, policy optimization, bandits, preference learning, or related methods
• Experience with post‑training methods for large models, including supervised fine‑tuning, reinforcement learning from feedback, direct preference optimization, reward modeling, or model adaptation
• Experience with distributed training, large-scale optimization, and multi-node or multi‑accelerator execution

Job Family

Research Development (RD)

Job Profile

Computer Science 2

Worker Type

Regular

Time Type

Full time

The expected hiring range for this position is $94,486.00 - $147,398.94.

Please note that the pay range information is a general guideline only. The pay offered to a selected candidate will be determined based on factors such as, but not limited to, the scope and responsibilities of the position, the qualifications of the selected candidate, business considerations, internal equity, and external market pay for comparable jobs. Additionally, comprehensive benefits are part of the total rewards package.

As an equal employment opportunity employer, and in accordance with our core values of impact, safety, respect, integrity and teamwork, Argonne National Laboratory is committed to a safe and welcoming workplace that fosters collaborative scientific discovery and innovation. Argonne encourages everyone to apply for employment. Argonne is committed to nondiscrimination and considers all qualified applicants for employment without regard to any characteristic protected by law.

Argonne employees, and certain guest researchers and contractors, are subject to particular restrictions related to participation in Foreign Government Sponsored or Affiliated Activities, as defined and detailed in United States Department of Energy Order 486.1A. You will be asked to disclose any such participation in the application phase for review by Argonne's Legal Department.

All Argonne offers of employment are contingent upon a background check that includes an assessment of criminal conviction history conducted on an individualized and case-by-case basis. Please be advised that Argonne positions require upon hire (or may require in the future) for the individual be to obtain a government access authorization that involves additional background check requirements. Failure to obtain or maintain such government access authorization could result in the withdrawal of a job offer or future termination of employment.