AI Software Engineer
职位介绍
You will optimize and develop deep learning frameworks for GPU platforms, focusing on training and inference performance across multi-GPU and multi-node deployments.
Develop and optimize GPU kernels for AI operations and design models that maximize GPU throughput.
Collaborate with internal library teams and open-source maintainers to integrate optimizations upstream.
Work across the full deep learning pipeline, including graph compilers and compiler-driven performance improvements.
Tackle scale-out and scale-up workloads using cutting-edge compiler technologies and low-level toolchains (HIP, CUDA, ASM, CK, CUTLASS, Triton).
Apply software engineering best practices in Linux environments to deliver robust, maintainable solutions.
Develop and optimize GPU kernels for AI operations and design models that maximize GPU throughput.
Collaborate with internal library teams and open-source maintainers to integrate optimizations upstream.
Work across the full deep learning pipeline, including graph compilers and compiler-driven performance improvements.
Tackle scale-out and scale-up workloads using cutting-edge compiler technologies and low-level toolchains (HIP, CUDA, ASM, CK, CUTLASS, Triton).
Apply software engineering best practices in Linux environments to deliver robust, maintainable solutions.
查看完整职位
工作职责、任职要求、技能与福利 — 免费创建账号即可查看。
已有账户? 登录
相似职位
其他可能适合您的职位。
远程办公No
城市Shanghai, 中国