AI Technical Specialist

Key Skills

AI Robotic systems, C Programming, C++, CUDA/GPU programming, Keras software library, Large Language Models - LLMs, Machine learning techniques, Natural Language Processing (NLP), Natural Language Understanding-NLU, ONNX, OpenCL, Optimization, Python Programming, PyTorch, Recommendation Systems, TensorFlow, Triton

Job Description

WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world’s most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives.

AMD together we advance_

The Role

AMD is seeking an experienced machine learning developer to engage our customers, partners, and business teams and support solutions that accelerate adoption of AMD Pervasive AI products. Successful candidates will be responsible as a trusted advisor for the analysis and support of customer machine learning workloads and proof-points that demonstrate optimized implementation within specific application segments of AMD’s AI portfolio of products. In this capacity, you will help support the business for the expansion of AMD’s Pervasive AI strategy.

You will engage and provide product and application domain expertise and support directly with technical, business and sales teams working with customer applications. The other responsibilities will include implementation of machine learning models, applications and demos on AMD products and provide domain expertise and advice on mapping and implementation of machine learning applications across the AMD AI product portfolio.

As a member of the Developer Acceleration Team within AMD’s AI Product Management organization, you will have opportunity to work closely with internal stakeholders across the company as well as engage directly with external customers. Close interaction with Engineering, Applications, and Business Unit stakeholders is essential.

The Person

The type of person who will be successful in this role is highly collaborative, and willing to trailblaze and champion new ideas. You will exhibit excellent written and verbal communication skills and you will be highly analytical. Applicants should possess a MS/PhD in EE or equivalent. The position requires comprehensive knowledge of machine learning architectures, models, and developer tools. Prior work experience in accelerated machine learning is required.

Key Responsibilities:

• Support and deliver technical expertise to support customer success in deployments of machine learning solutions on AMD products
• Providing domain expertise in AMD silicon, IP, and software including compilers, libraries, and profiling tools
• Provide domain expertise in foundational machine learning architectures, tools, and developer flows
• Implementation and training of machine learning models in common frameworks including PyTorch, Keras/Tensorflow, ONNX
• Experience with deployed production runtime serving of machine learning applications
• Experience publishing to open source, including github repositories and industry collaborations
• Experience with HIP, OpenAI Triton, OpenCL, SYCL, Cuda or other kernel development programming flows
• Developing in Python, C/C++, within Jupyter notebooks
• End to end performance testing and optimization of machine learning workloads
• Application profiling with CPU and accelerator tools (VTune, NVprof, etc)
• Integration and development of demonstration/application software

Preferred Experience in following skills

• Experience with foundational machine learning model architecture development or implementation including LLMs, Embeddings, Graph Neural Networks, CNNs, Stable Diffusion, ASR, and NLP pipelines
• Domain expertise in one or more of the following application domains such as Computer Vision (CV), Robotics, Smart City, Industrial Automation, Autonomous Driving, Natural Language Understanding (NLU/NLP), Graph Analytics, Recommendation Systems, Anomaly Detection, and High Performance Computing (HPC)
• Perl/Python/C++ scripting for test automation
• Effective communication and problem-solving skills
• Motivating leader with good interpersonal skills

Academic Credentials

• Bachelors, Masters, or PhD (Computer Science, Computer Engineering, Electrical Engineering, or related equivalent)