AI Models GPU Deployment 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 enable and optimize key AI models (LLMs, Vision, MultiModal) on AMD GPUs.
Tuning and accelerating AI frameworks (PyTorch, TensorFlow) in upstream repositories.
Collaborate with internal GPU library teams to analyze training and inference pipelines.
Work with open-source framework maintainers to integrate changes upstream.
Optimize GPU kernels and AI operators (GEMM, Attention) for scale-up (multi-GPU) and scale-out (multi-node) deployments.
Apply software engineering best practices—debugging, testing, and performance analysis—to deliver robust solutions.
Tuning and accelerating AI frameworks (PyTorch, TensorFlow) in upstream repositories.
Collaborate with internal GPU library teams to analyze training and inference pipelines.
Work with open-source framework maintainers to integrate changes upstream.
Optimize GPU kernels and AI operators (GEMM, Attention) for scale-up (multi-GPU) and scale-out (multi-node) deployments.
Apply software engineering best practices—debugging, testing, and performance analysis—to deliver robust 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.