Deep Learning Performance Software Intern - 2027
Salary not listed • China, Shanghai • Internship
Posted 1mo ago
Remote Work Policy
On-site
Employment Type
Internship
Categories
AI Infrastructure Engineer
About the job
We are expanding our research and development for deep learning and are seeking excellent Software Engineers to join our team. We specialize in developing GPU-accelerated deep learning software, enabling breakthroughs in numerous areas. Join the team that builds software to enable new solutions. Your ability to work in a fast-paced, customer-oriented team is required, and excellent communication skills are necessary.
Responsibilities
- Create and maintain SKILL, Wiki, and agent harness.
- Develop TileGym, Triton CUDA TileIR backend, and CUDA Tile.
- Develop highly optimized deep learning kernels through a tile-based GPU programming model.
- Perform end-to-end performance optimization using a tile-based GPU programming model.
- Conduct performance optimization, analysis, and tuning.
Requirements
- Currently pursuing a degree in an engineering or computer science related field; Master's or Doctoral candidate preferred.
- Understanding of 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 is a plus.
- MLIR experience is a plus.
- Knowledge of performance modeling, profiling, debugging, code optimization, or architectural knowledge of CPU and GPU.
- GPU programming experience (CUDA or OpenCL) is desired.