Remote Research Engineer Intern - AI Systems

Description:

• Internship position for a Research Engineer Intern focused on AI Systems, Compiler and Runtime Optimization, and GPU/Accelerator Kernel Development.
• Work on a project involving Trainium, GPU kernels, and LLM systems optimization over a 12–16 week period, impacting AI application performance.

Requirements:

• Currently pursuing a BS, MS, or PhD in Computer Science, Computer Engineering, or a related field.
• Solid programming skills in Python and familiarity with C++.
• Understanding of GPU/accelerator architecture fundamentals.
• Experience writing CUDA, Triton, ROCm/HIP, or Neuron kernels.
• Strong understanding of AI frameworks and profiling tools.
• Strong problem-solving skills and ability to work independently in a remote environment.

Benefits:

• Opportunity to work on cutting-edge AI infrastructure problems with access to advanced hardware.
• Direct mentorship from experienced engineers.
• Competitive compensation and flexible remote work environment.
• Fast track to a full-time return offer for top performers.