GPU Kernel Developer - AI Models Advanced Micro Devices, Inc

3.0 - 7.0 years

0 Lacs

karnataka

On-site

As a GPU Kernel Developer specializing in AI Models at AMD, you will play a crucial role in developing high-performance kernels for cutting-edge GPU hardware. Working alongside a team of industry specialists, you will have the opportunity to leverage your expertise in GPU computing and hardware architecture to optimize AI and HPC applications. Your contributions will be instrumental in advancing key AI operators on AMD GPUs, ensuring optimal performance and efficiency. Your responsibilities will include developing high-performance GPU kernels for AI operators, optimizing GPU code through structured methodologies, and supporting critical workloads in NLP, LLM, Recommendation, Vision, and Audio domains. Collaboration with system architects, GPU specialists, and performance validation teams will be essential to analyze and enhance training and inference processes for AI. Additionally, you will engage with open-source framework maintainers to integrate code changes and drive AI operator performance improvements. To excel in this role, you should possess a strong background in GPU kernel development, proficiency in HIP, CUDA, OpenCL, and Triton programming, and a deep understanding of GPU hardware. Your expertise in profiling, debugging tools, and software engineering best practices will be key in optimizing GPU kernels and enhancing performance. A Master's or PhD in Computer Science, Computer Engineering, or a related field is preferred to support your technical skills and knowledge. Join AMD in revolutionizing next-generation computing experiences through innovative GPU technologies and contribute to solving the world's most critical challenges. Be a part of our mission to build exceptional products that shape the future of data centers, artificial intelligence, PCs, gaming, and embedded systems. Together, we advance the boundaries of innovation and drive impactful change in the industry and beyond.,

Posted 2 weeks ago

Apply

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.