AI Frameworks Intern

About the role

This internship supports enablement of a deep learning framework's GPU backend, including implementing and debugging operators and kernels in Python and C++, fixing platform build and test issues, and contributing to upstream open-source work. Applicants should be pursuing a computer science or related degree, with solid Python, basic C/C++, familiarity with ML frameworks, and Git experience.