Research Engineer Intern - AI Systems

Remote Remote · United States

pythonawspytorch engineeringintern

ashby

Match & cover letter

Create a profile to see your match and get a cover letter for this role.

Create profile and match

Job description

You will work on optimizing AI systems and GPU kernels during a remote internship. This role offers the opportunity to contribute to real projects that impact AI applications.

Details

Remote work available globally
Internship position
Currently pursuing a BS, MS, or PhD in Computer Science or related field
Skills required: Python, CUDA, PyTorch, AWS

The work

Implement and optimize compute kernels for AI models
Build custom operators using CUDA and PyTorch
Profile and improve inference performance in custom runtimes
Ship code to open-source AI infrastructure projects