GPU Machine Learning Software Developer- 104422

Company: Advanced Micro Devices
Location: Austin, Texas, United States
Type: Full-time
Posted: 13.AUG.2021


What you do at AMD changes everything At AMD, we push the boundaries of what is possible. We believe in changing the world for the better by...


What you do at AMD changes everything At AMD, we push the boundaries of what is possible. We believe in changing the world for the better by driving innovation in high-performance computing, graphics, and visualization technologies - building blocks for gaming, immersive platforms, and the data center. Developing great technology takes more than talent: it takes amazing people who understand collaboration, respect, and who will go the "extra mile" to achieve unthinkable results. It takes people who have the passion and desire to disrupt the status quo, push boundaries, deliver innovation, and change the world. If you have this type of passion, we invite you to take a look at the opportunities available to come join our team.GPU Machine Learning Software Developer The Role: Do you want to develop a GPU kernel generator to deliver the highest performance machine learning tensor operations with support for layer fusion? AMD is searching for hardworking and motivated computer scientists and engineers to implement machine learning primitives as part of the AMD Radeon Open Ecosystem (ROCm) . The Person: You are accustomed to working in a dynamic, geographically distributed agile team, where partnership and collaboration are paramount. You possess excellent written and verbal communication skills, strong attention to detail, and the ability to express your work in a clear, cohesive fashion. You are results-oriented and accustomed to tight deadlines and changing priorities. Most importantly, you are constantly thinking of ways to improve performance of software and hardware. Key Responsibilities: Design, implement, and test a GPU kernel generator for tensor operations like matrix multiplication and convolutions used in a variety of high-performance machine learning libraries and frameworks Benchmark, analyze, and optimize performance of key machine learning applications and participate in the co-design across AMD's ML hardware and software stack Communicate and collaborate with key technical specialists across AMD and with our partners and customers to improve ROCm applications, libraries, and tools, as well as hardware Deliver high-quality code and documentation following best practices for open-source software development Apply software engineering methodologies to ensure robust, maintainable code Preferred Experience: Strong background developing applications and libraries in C++, especially high-performance computing and/or scientific software Understanding of linear algebra, especially from an implementation perspective [QS1] GPU software development using HIP, CUDA, or OpenCL Understanding of CPU and GPU architectures including memory hierarchy, instruction scheduling, and performance tradeoffs Experience with low-level optimization techniques including assembly programming and/or vectorization Experience developing code generators or compilers In-depth knowledge of best-practices in software development, including testing, profiling, debugging, documentation, version control, issue tracking, and planning Academic Credentials: B.Sc. or B.Eng. degree in Computer Science, Software Engineering, Electrical Engineering, Applied Mathematics, or equivalent Advanced degrees, such as M.Sc., M.Eng., Ph.D. are preferred. Location: Austin, TX #LI-JG1Requisition Number: 104422 Country: United States State: Texas City: Austin Job Function: Design AMD does not accept unsolicited resumes from headhunters, recruitment agencies or fee based recruitment services. AMD and its subsidiaries are equal opportunity employers. We consider candidates regardless of age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status. Please click here for more information.

Apply Now


Loader2 Processing ...