MPI4AI is a multi-university collaboration between Tennessee Technological University, the University of Tennessee, Knoxville, Stony Brook University, and the Illinois Institute of Technology. It aims to specialize MPI collective communication patterns for the needs of modern AI workloads running at scale on HPC systems, including training and inferences using various levels of parallelism (data, tensor, pipeline, sequence), agentic collaboration, neural architecture search. Aspects such as how to relax consistency (e.g. ignore stragglers), integrate with heterogeneous memory tiers (GPU HBM, host memory, SSDs, external storage) and how to enable asynchronous, GPU-centric communication are crucial in this context and will be advocated for the upcoming MPI-5 and MPI-6 standards. The student's role in the project will be to co-design communication patterns specifically tailored for AI applications (training, inference, agentic workflows) and to identify novel ways to apply them in AI runtimes (Torch, vLLM, DeepSpeed, etc)
Education and Experience Requirements
o Currently enrolled in undergraduate or graduate studies at an accredited institution.
o Graduated from an accredited institution within the past 3 months; or
o Actively enrolled in a graduate program at an accredited institution.
Job Family
Visiting Student GraduateJob Profile
Visiting Student - GraduateWorker Type
Contingent WorkerTime Type
Full timeScheduled Weekly Hours
40EEO Information
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.
| Location | Lemont, Illinois |
Browse dozens of recruiter approved resume templates, layouts and formats. Choose your favorite and make it your own in minutes.
Free resume templatesImprove your existing resume or start from scratch and create a standout, ATS-friendly resume. Add job-specific content, download and apply.
Free resume builder