Argonne National Laboratory

Software Engineer - AI Inference for Science

Argonne National Laboratory$94K — $147K *
Information Technology
5 - 7 years of experience
Job Overview by Ladders

Qualifications

  • Overall familiarity with AI models including deployment and resource requirements.
  • Experience with AI hosting frameworks like vLLM or SGLang.
  • Proficiency in REST API development using Python frameworks like FastAPI.
  • Strong programming skills in languages such as Python or C/C++.
  • Experience with version control systems, especially git.
  • Ability to create and maintain high-quality software.

Responsibilities

  • Explore and engineer solutions for AI inference in scientific workflows.
  • Integrate AI model usage within scientific applications through programmatic access.
  • Submit large batches of prompts for parallel processing on HPC systems.
  • Work collaboratively with application teams and other stakeholders.
  • Adapt and deploy new AI models as technology evolves.
  • Contribute to scientific initiatives solving complex problems.

Benefits

  • Hybrid remote work with flexibility of onsite and remote days.
  • Engagement with diverse science and industry teams.
  • Opportunities for contributing to major scientific initiatives.
  • Comprehensive employee benefits package available.
Full Job Description
The ALCF has an opening for a Software Engineer working in the space of enabling AI for science, specifically targeting scalable inference leveraging HPC systems and AI accelerators. The successful candidate will join the Data Services and Workflows group, which focuses on scientific workflows that combine large-scale data, simulations, analysis, and AI. In this position, the candidate can expect to explore and engineer solutions for AI inference integrated within scientific workflows, via programmatic access using standard programming interfaces (e.g. OpenAI API), and through submission of large batches of prompts for parallel processing; both scenarios require efficient execution on underlying resources, including ALCF's HPC systems and AI testbed machines. This position demands a good understanding of currently available AI models (LLMs and otherwise), their compute and memory requirements, and how to utilize the underlying hardware for high responsiveness. As the space of AI models evolves, we will adapt and deploy new models and functionality.

The Data Services and Workflows group--and this position--involves work in a highly collaborative environment involving science application teams, academia and industry, as well as other national labs and agencies, to solve some of the world's largest and most complex problems in science and engineering. The candidate will engage with science application teams and contribute to broader scientific initiatives.

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

Position Requirements

Required skills and qualifications
  • Overall familiarity with AI model use, deployment, and memory- and compute-requirements.
  • Familiarity with one or more AI hosting frameworks such as vLLM or SGLang.
  • Familiarity with REST API development in Python using frameworks such as FastAPI.
  • Comprehensive experience programming in one or more programming languages such as Python, C/C++.
  • Ability to create, maintain, and support high-quality software is essential.
  • Work with and contribute to domain-specific software and models.
  • Experience with version control software such as git.
  • Ability to work collaboratively in a fast-paced environment.
  • Effective written and oral communications skills.
  • Ability to model Argonne's core values of impact, safety, respect, integrity and teamwork.


Preferred skills and qualifications
  • Experience designing or operating distributed inference or data services, including request routing, asynchronous execution, queueing, fault tolerance, and performance monitoring.
  • Experience integrating services with HPC schedulers (e.g., Slurm, PBS), including resource provisioning, job lifecycle management, and balancing latency-sensitive and throughput-oriented workloads.
  • Experience optimizing AI inference performance (e.g., batching, memory management, model parallelism, quantization, accelerator utilization) on GPU- or accelerator-based systems.
  • Familiarity with secure, multi-user services, including authentication/authorization, API security, and operating within institutional or regulated environments.
  • Experience with log aggregation, analytics, and observability (familiarity with Prometheus, Grafana, log aggregators, OLAP or data warehousing techniques).


This position can be hired at one of two levels (RD2 or RD3) and the requirements for each are as follows:
  • RD2: Bachelor's degree and 5+ years of experience, Master's degree and 3+ years of experience, or PhD, or equivalent. The expected pay range for this level is between $94,486 - $147,399.
  • RD3: Bachelor's degree and 8+vyears of experience, Master's degree and 5+ years of experience, or PhD and 4+ years of experience, or equivalent. The expected pay range for this level is $116,250 - $181,350.


Job Family
Research Development (RD)

Job Profile
Software Engineering 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.

Click here to view Argonne employee benefits!

About Argonne National Laboratory

Argonne National Laboratory is a science and engineering research national laboratory operated by the University of Chicago Argonne LLC for the United States Department of Energy. It is located in Lemont, Illinois, outside of Chicago. Argonne conducts research in a variety of fields, including energy, environment, national security, and technology. The laboratory was founded in 1946 as part of the Manhattan Project and has since become one of the largest science and engineering research laboratories in the United States.
Learn more about Argonne National Laboratory
Size
3,400 employees
Industry
Founded
1946

Similar Jobs

More Jobs at Argonne National Laboratory

More Information Technology Jobs

Find similar Software Engineer - AI Inference for Science jobs: