Sr. Staff AI/ML Development Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Advanced Microdevices Pvt. Ltd. (India) ยท 12 hours ago

Sr. Staff AI/ML Development Engineer

Advanced Micro Devices, Inc is a leading company in the computing industry focused on innovation and collaboration. They are seeking a Senior Staff AI/ML Development Engineer to lead efforts in advancing AI inference and training performance on AMD GPU platforms, working with cutting-edge models and optimizing performance.

BiopharmaBiotechnologyIndustrialManufacturing

Responsibilities

Follow the latest developments in AI/ML models and techniques through publications, blog posts, conferences, etc
Develop innovative new techniques that leverage the strength of AMD platforms for improved performance and higher quality and share them with the AI community through publications and open-source contributions
Drive revenue growth by partnering with customers to optimize their AI workloads on AMD platforms
Develop, maintain, and enhance both internal and open-source tools for profiling, analyzing, and projecting performance
Identify strategic opportunities for AMD in emerging AI technologies
Communicate findings with AMD internal teams and, as appropriate, the external AI community

Qualification

PythonPytorchJAXAI model optimizationNLPGenerative AIDistributed systemsLow-level GPU programmingPerformance optimizationCommunication skills

Required

Knowledge of, experience with, and passion for both AI/ML models/applications and hardware/software systems
Ability to communicate findings with AMD internal teams and, as appropriate, the external AI community
M.S. or PhD in Computer Science, Computer Engineering, Electrical Engineering, or a related field. B.S. with additional experience will also be considered

Preferred

Experience developing and debugging in Python and ML frameworks such as Pytorch and JAX
Experience with current state-of-the-art AI models, especially in NLP and Generative AI
Experience optimizing the performance of AI models on accelerator hardware
Understanding of AI model optimization techniques such as operator fusion, quantization, and sparsity
Solid understanding of hardware & software systems, including distributed systems
Ability to analyze and address performance bottlenecks
Low-level GPU programming and performance optimization experience a plus

Benefits

AMD benefits at a glance.

Company

Advanced Microdevices Pvt. Ltd. (India)

twittertwittertwitter
company-logo
Advanced Microdevices (mdi) is a leader in innovative membrane technologies.

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
Nalini Kant Gupta
Founder & Managing Director
Company data provided by crunchbase