Machine Learning Research Engineer Co-op

Job Summary

At Red Hat we believe the future of AI is open and we are on a mission to bring the power of open-source LLMs and distributed LLM inference to every enterprise. We are seeking a highly motivated research intern to join our Machine Learning Research Team. As a research intern, you will work on cutting-edge networking techniques for ML workloads and contribute to research and engineering efforts that make distributed LLM inference faster, efficient, and more accessible. This is an exciting opportunity to gain hands-on experience in applied networking for ML while working with leading experts in the field.

Responsibilities

• Research via experimentation and theoretical modeling the network bandwidth requirements and trade-offs in Prefill-Decode (P/D) disaggregated LLM serving. 

• Research and implement networking techniques/methods for high-performance KV cache transfers in deployment setups without RDMA networking.

• Conduct experiments to evaluate the impact of newly developed non-RDMA KV Cache transfer techniques on performance (latency and throughput) in P/D LLM serving. 

• Collaborate with researchers and engineers to integrate the networking techniques/methods into real-world distributed inference workflows (e.g. in llm-d)

• Document findings and contribute to technical reports, research theses, blog posts, or research publications.

Requirements

• Currently pursuing a Masters (with research) or Ph.D. degree in Computer Science, Electrical Engineering, Machine Learning, or a related field.

• Strong programming skills in C/C++, Rust, and Python. 

• Experience with the Linux network stack including frameworks such as DPDK or eBPF/XDP.

• Strong analytical and problem-solving skills.

• Excellent communication skills and ability to work in a team-oriented research environment.

• Familiarity with distributed LLM serving with prefill/decode disaggregation and KV cache transfers is a plus, but not required.

Why work with us

• Hands-on experience with state-of-the-art systems for ML research.

• Mentorship from leading experts in LLM inference and networking.

• Opportunity to contribute to research papers, patents, or open-source projects.

• Competitive stipend and potential for full-time opportunities.

About Red Hat

Red Hat is the world’s leading provider of enterpriseopen source software solutions, using a community-powered approach to deliver high-performing Linux, cloud, container, and Kubernetes technologies. Spread across 40+ countries, our associates work flexibly across work environments, from in-office, to office-flex, to fully remote, depending on the requirements of their role. Red Hatters are encouraged to bring their best ideas, no matter their title or tenure. We're a leader in open source because of our open and inclusive environment. We hire creative, passionate people ready to contribute their ideas, help solve complex problems, and make an impact.

Inclusion at Red Hat
Red Hat’s culture is built on the open source principles of transparency, collaboration, and inclusion, where the best ideas can come from anywhere and anyone. When this is realized, it empowers people from different backgrounds, perspectives, and experiences to come together to share ideas, challenge the status quo, and drive innovation. Our aspiration is that everyone experiences this culture with equal opportunity and access, and that all voices are not only heard but also celebrated. We hope you will join our celebration, and we welcome and encourage applicants from all the beautiful dimensions that compose our global village.

Equal Opportunity Policy (EEO)
Red Hat is proud to be an equal opportunity workplace and an affirmative action employer. We review applications for employment without regard to their race, color, religion, sex, sexual orientation, gender identity, national origin, ancestry, citizenship, age, veteran status, genetic information, physical or mental disability, medical condition, marital status, or any other basis prohibited by law.

Red Hat does not seek or accept unsolicited resumes or CVs from recruitment agencies. We are not responsible for, and will not pay, any fees, commissions, or any other payment related to unsolicited resumes or CVs except as required in a written contract between Red Hat and the recruitment agency or party requesting payment of a fee.

Red Hat supports individuals with disabilities and provides reasonable accommodations to job applicants. If you need assistance completing our online job application, email application-assistance@redhat.com. General inquiries, such as those regarding the status of a job application, will not receive a reply.