Work with AI agents to create high performance low-level kernels in a Domain-Specific language for
Ensure functional correctness and numerical accuracy of kernels
Conduct performance analysis of the kernels to identify performance bottlenecks and areas that need
Develop a custom harness to recursively improve kernel generation and remove the need for human

🎯 Requirements

Currently enrolled in an MS or PhD program in Computer Science, Computer Engineering, Computational
Demonstrated track record of using AI agents and harnesses to improve coding performance
Working knowledge of key kernels for transformer inference including performance analysis and
Familiarity with one or more low-level ML accelerator DSLs (CUDA, Triton, Gluon, etc.)

🎁 Benefits

Interns are eligible to participate in the Company’s generous benefits programs, subject to

Apply on employer's website

This employer gathers applications via their own applicant tracking system.

You will be redirected to an external application form.

Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.

Activate JobCopilot