Advanced Microdevices Pvt. Ltd. (India) · 2 days ago
AI Models GPU deployment software Engineer
Austin, TX
Full-time
Onsite
Mid LevelAdvanced Micro Devices, Inc is a company focused on accelerating next-generation computing experiences across various domains including AI and data centers. They are seeking a software engineer to optimize AI models on AMD GPUs and enhance application performance through collaboration with internal teams and open-source framework maintainers.
BiopharmaBiotechnologyIndustrialManufacturing
Responsibilities
Enable and optimize key AI models (LLM, Vision, MultiModal, etc.) on AMD GPUs
Optimize AI frameworks like PyTorch, TensorFlow, etc. on AMD GPUs in upstream open-source repositories
Collaborate and interact with internal GPU library teams to analyze and optimize training and inference for AI
Work with open-source framework maintainers to understand their requirements – and have your code changes integrated upstream
Optimize GPU kernels, understand and drive AI operator performance (GEMM, Attention, etc.) with specialized teams
Work in a distributed computing setting to optimize for both scale-up (multi-GPU) and scale-out (multi-node) systems
Apply your knowledge of software engineering best practices
Qualification
Required
Strong technical and analytical skills in C++/Python AI development in a Linux environment
Ability to work as part of a team, while also being able to work independently, define goals and scope and lead your own development effort
Enable and optimize key AI models (LLM, Vision, MultiModal, etc.) on AMD GPUs
Optimize AI frameworks like PyTorch, TensorFlow, etc. on AMD GPUs in upstream open-source repositories
Collaborate and interact with internal GPU library teams to analyze and optimize training and inference for AI
Work with open-source framework maintainers to understand their requirements – and have your code changes integrated upstream
Optimize GPU kernels, understand and drive AI operator performance (GEMM, Attention, etc.) with specialized teams
Work in a distributed computing setting to optimize for both scale-up (multi-GPU) and scale-out (multi-node) systems
Apply your knowledge of software engineering best practices
Masters or PhD or equivalent experience in Computer Science, Computer Engineering, or related field
Preferred
Knowledge of GPU computing (HIP, CUDA, OpenCL)
AI model experience or knowledge - Natural Language Processing, Vision, Audio, Recommendation systems
Excellent C/C++/Python programming and software design skills, including debugging, performance analysis, and test design
Experiences to run workloads on large scale heterogeneous cluster is a plus
Experiences to optimize GPU kernels for performance is a plus
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