Staff Scientist: Post-Training & RL for Scientific AI

Argonne National Laboratory; Lemont (IL); Hybrid; USD 94,486 - 147,398; Full time

Job summary

Argonne National Laboratory in Lemont, Illinois, is seeking a Staff Scientist to develop, scale, and evaluate post-training methods for AI models. The role involves interdisciplinary collaboration and aims to advance machine learning techniques for scientific applications.

The ideal candidate will have a strong foundation in machine learning, programming, and mathematical optimization, and will contribute to research that impacts the AI for science mission.

Qualifications

• 5+ years of experience with a Bachelor's; 3+ years with a Master's; or PhD.
• Advanced knowledge in machine learning and statistics is essential.
• Strong background in mathematical optimization and numerical methods.

Responsibilities

• Conduct research aligned with computational AI and scientific discovery.
• Develop post-training methods for scientific foundation models.
• Design methods for applying reinforcement learning to large-scale environments.

Skills

Machine learning

Reinforcement learning

Large-scale model training

Python

C

C++

Mathematical optimization

Education

Bachelor’s degree or Master's or PhD in relevant fields

Tools

PyTorch

JAX

Job description

Argonne National Laboratory in Lemont, Illinois, is seeking a Staff Scientist to develop, scale, and evaluate post-training methods for AI models. The role involves interdisciplinary collaboration and aims to advance machine learning techniques for scientific applications.

The ideal candidate will have a strong foundation in machine learning, programming, and mathematical optimization, and will contribute to research that impacts the AI for science mission.