Luxoft is looking for an AI software development engineer to develop ML kernels in the Triton kernel language. We are looking for an engineer who is passionate about optimizing Machine Learning GPU kernels and improving the performance of key applications and benchmarks. What you do directly impacts the performance of AMD GPUs and enables us to become a competitive solution for generative AI. Become a part of our high-impact and incredibly talented Triton kernels team.

• Develop ML kernels for matrix multiplication, Flash Attention and other ML operators
• Benchmark, perform competitive analysis and optimize kernels to improve performance
• Collaborate with the GPU architecture team to improve future generations
• Apply knowledge of software engineering best practices