The Research Engineer Intern will develop and optimize compute kernels for AI hardware and improve inference performance for large language models. They will also build benchmarks, profile system performance, and contribute code to open-source AI infrastructure projects.
Design and implement high-performance kernels for AI operations across NVIDIA, AMD, and AWS Trainium hardware. Develop scalable distributed training and inference solutions to optimize the performance of Large Language Model infrastructure.
