Graduate (3-month) Intern – AI Data Center Workloads Characterization & Simulation
- Nrel
- Golden, CO
- 1w ago
- Full-Time
- Remote
.
.
.
.
Join the National Laboratory of the Rockies (NLR), where world-class scientists, engineers, and experts are accelerating energy innovation through breakthrough research and systems integration. From our mission to our collaborative culture, NLR stands out in the research community for its commitment to an affordable and secure energy future. Spanning foundational science to applied systems engineering and analysis, we focus on solving complex challenges to deliver advanced, secure, reliable, and cost-effective energy solutions. Our work helps strengthen U.S. industries, support job creation, and promote national economic growth.
At NLR, you'll find a mission-driven environment supported by state-of-the-art facilities, multidisciplinary research teams, and strong collaborations with industry, academia, and other national laboratories. We offer robust professional development opportunities, and a competitive benefits package designed to support your career and well-being.
The AI, Learning and Intelligent Systems (ALIS) Group within the National Lab of the Rockies’ (NLR) Computational Science Center has an exciting opportunity for a graduate student to actively engage in a project developing an ecosystem of analysis and simulation tools for understanding power consumption and infrastructure requirements of AI data centers.
We are looking for a 3-month spring/summer student (remote, in-person, or hybrid). As a valued member of the team, the graduate student will collaborate closely with peers to improve NLR applications. The successful candidate will use software-based power measuring tools to sample the consumption of AI LLM workloads (fine-tuning, offline and online inference) on NLR’s supercomputing (HPC). We particularly welcome candidates with creative problem-solving skills, interest in cross-disciplinary collaboration, and a passion for NLR's mission.
Responsibilities include:
Contribute to the development of software to execute power characterization experiments of AI LLM workloads under different parameter assumptions i.e. sensitivity analysis
Performing computational experiments, running batch experiments on NLR HPC using SLURM
Analyzing results capturing trends and tradeoffs between parameter configurations
Coordinating data exchange to inform other team members’ efforts
Contribute to writing and presenting reports and conference proceedings related to characterization tests
.
* Must meet educational requirements prior to employment start date.
Currently enrolled in or recently graduated from a master’s degree, or currently enrolled in a PhD program in an Engineering, Computer Science, Applied Math, or related field
Demonstrated experience with software development in Python
Demonstrated experience with high-performance computing
Demonstrated experience with coding repositories i.e. Git
Demonstrated experience with conducting sensitivity analysis
Demonstrated experience with MLPerf and VLLM
Demonstrated experience in reading and writing scientific documents
Demonstrated experience with HPC i.e. Bash/Shell/Command line and SLURM
Demonstrated experience with AI LLM workloads, generative AI etc.
.
The anticipated closing window for application submission is up to 30 days and may be extended as needed.
NLR takes into consideration a candidate’s education, training, and experience, expected quality and quantity of work, required travel (if any), external market and internal value, including seniority and merit systems, and internal pay alignment when determining the salary level for potential new employees. In compliance with the Colorado Equal Pay for Equal Work Act, a potential new employee’s salary history will not be used in compensation decisions.
* Based on eligibility rules
NLR is committed to maintaining a drug-free workplace in accordance with the federal Drug-Free Workplace Act and complies with federal laws prohibiting the possession and use of illegal drugs. Under federal law, marijuana remains an illegal drug.
If you are offered employment at NLR, you must pass a pre-employment drug test prior to commencing employment. Unless prohibited by state or local law, the pre-employment drug test will include marijuana. If you test positive on the pre-employment drug test, your offer of employment may be withdrawn.
Please note that in order to be considered an applicant for any position at NLR you must submit an application form for each position for which you believe you are qualified. Applications are not kept on file for future positions. Please include a cover letter and resume with each position application.
.
All qualified applicants will receive consideration for employment without regard basis of age (40 and over), color, disability, gender identity, genetic information, marital status, domestic partner status, military or veteran status, national origin/ancestry, race, religion, creed, sex (including pregnancy, childbirth, breastfeeding), sexual orientation, and any other applicable status protected by federal, state, or local laws.
E-Verify is a registered trademark of the U.S. Department of Homeland Security. This business uses E-Verify in its hiring practices to achieve a lawful workforce.