Description
We are seeking a Deep Learning Performance Software Engineering Intern to join our team. You will be working on developing GPU-accelerated deep learning software.
Responsibilities:
- Creating and maintaining SKILL, Wiki, and agent harness
- Developing TileGym, Triton CUDA TileIR backend, and CUDA Tile
- Developing highly optimized deep learning kernels through tile-based GPU programming model
- End-to-end performance optimization through tile-based GPU programming model
- Performance optimization, analysis, and tuning
Requirements:
- Pursuing a degree in an engineering or computer science related field (masters or doctoral candidate preferred)
- Understands core components of agentic systems, including LLM APIs, prompting, tool use/function calling, agent loops, planning, reasoning, memory, RAG, skills, MCP, sub-agents, multi-agent architectures, context engineering, and harness engineering
- Excellent C/C++ programming and software design skills
- Python experience a plus
- MLIR experience a plus
- Performance modelling, profiling, debug, and code optimization or architectural knowledge of CPU and GPU
- GPU programming experience (CUDA or OpenCL) desired
This listing is enriched and indexed by YubHub. To apply, use the employer's original posting:
https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/China-Shanghai/Deep-Learning-Performance-Software-Intern---2027_JR2025966