AMD
Staff AI Frameworks Engineer (C/C++/Python) — GPU-ML
AMD, San Jose, California, United States, 95199
A semiconductor company in San Jose is seeking a Staff Software Development Engineer specializing in AI/ML frameworks to optimize and enhance frameworks for AMD GPUs. This mid-senior level position requires proficiency in C++ and Python, alongside expertise in AI software frameworks like PyTorch. The role focuses on developing GPU kernels, optimizing deep learning models, and collaborating with both internal teams and open-source maintainers. Candidates should hold at least a BS in relevant fields and have experience in GPU optimization.
#J-18808-Ljbffr