AI Software Engineer
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
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.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
Already have an account? Log in
Similar openings
Other roles that could suit you.