Advanced Microdevices Pvt. Ltd. (India) · 7 hours ago
GPU KERNEL OPTIMIZATION ENGINEER
Austin, TX
Full-time
Hybrid
Senior LevelAdvanced Micro Devices, Inc is committed to building innovative products that enhance computing experiences across various domains. They are seeking a talented GPU Kernel Optimization Engineer to develop and optimize high-performance GPU kernels for machine learning applications, collaborating with teams to improve performance and functionality.
BiotechnologyIndustrialPharmaceuticalManufacturingBiopharma
No H1B
Responsibilities
Analyze the performance of GPU kernels and provide insight on optimization paths
Generate new kernels or optimize existing kernels for matrix multiplication
Communicate and collaborate with key technical experts across AMD and with our partners and customers to improve ROCm applications, libraries, and tools, as well as hardware
Contribute to the design and implementation of AMD’S GPU kernel generator used in a variety of high-performance machine learning libraries and frameworks
Develop optimization and analysis passes on graph representations of GPU kernels
Develop analytical performance models and heuristics for kernel selection
Deliver high-quality code and documentation following best practices for open-source software development
Apply software engineering methodologies to ensure robust, maintainable code
Menor junior engineers and develop technical relationships with peers and partners
Qualification
Required
Analyze the performance of GPU kernels and provide insight on optimization paths
Generate new kernels or optimize existing kernels for matrix multiplication
Communicate and collaborate with key technical experts across AMD and with our partners and customers to improve ROCm applications, libraries, and tools, as well as hardware
Contribute to the design and implementation of AMD'S GPU kernel generator used in a variety of high-performance machine learning libraries and frameworks
Develop optimization and analysis passes on graph representations of GPU kernels
Develop analytical performance models and heuristics for kernel selection
Deliver high-quality code and documentation following best practices for open-source software development
Apply software engineering methodologies to ensure robust, maintainable code
Menor junior engineers and develop technical relationships with peers and partners
PhD or Master's degree in Applied Mathematics, Computer Science, Computer Engineering, Electrical Engineering, or equivalent
Preferred
Demonstrate strong knowledge and understanding of GPU architecture
Hands-on professional experience writing and optimizing GPU code
Experience with GEMM algorithms and optimization
Strong background developing applications and libraries in modern C++ and Python
Strong communication skills and work ethics
Experience with best practices in software development, including testing, profiling, debugging, documentation, version control, and issue tracking
Experience with low-level optimization techniques; assembly programming and/or vectorization
Ability to closely interact with technical leads, program managers, and test teams to maintain and release production software
Experience with agile software development practices
Benefits
AMD benefits at a glance.
Company
Advanced Microdevices Pvt. Ltd. (India)
Advanced Microdevices (mdi) is a leader in innovative membrane technologies.
Founded in 1976
501-1000 employees
Funding
Current Stage
Late StageLeadership Team
Nalini Kant Gupta
Founder & Managing Director
Recent News
2024-10-18
2024-10-16
Company data provided by crunchbase