Assistant Computational Scientist, Argonne National Laboratory
I work in supporting the AI/ML software stack on Aurora. My other research effort focuses on understanding the performance portability of AI/ML workloads for distributed training and inference, with particular attention to collective communication. Recent efforts include exploring kernel engineering through Triton within the PyTorch ecosystem.