Apply on Employer Site

Advanced Microdevices Pvt. Ltd. (India) · 2 months ago

Sr. Staff Engineer, AI Models and Applications

San Jose, CA

Full-time

Hybrid

Senior Level, Lead/Staff

Advanced Micro Devices, Inc is a company focused on building products that enhance computing experiences across various domains. They are seeking a passionate Sr. Staff Engineer to join their AI Models and Applications team to innovate and optimize Generative AI training and inference at scale.

BiopharmaBiotechnologyIndustrialManufacturing

Responsibilities

Propose and apply innovative techniques to support both training and inferencing including innovative transformer architectures, parallelism strategies to train on large clusters, low-precision training

Implement novel efficient architectures for Generative AI models for training and inference and showcase benefits on AMD

Work with open-source framework and community (e.g., PyTorch, JAX, Hugging Face) to integrate AMD optimized models, libraries and publish training recipes

Collaborate with software and hardware team to E2E co-optimize performance on current and future AMD solutions

Publish and promote your work within AMD and at external venues

Qualification

Generative AI modelsDeep learning frameworksAlgorithmic innovationLow-precision trainingParallelism strategiesPublications in AISoftware developmentCommunicationTeam collaboration

Required

Deep technical understanding of the latest generative AI applications like large language models (LLMs), large multimodal models (LMMs), image/video generation

Experience training models at scale

Passionate about innovating efficient approaches to enable distributed training and inference at scale on AMD devices

Implement novel efficient architectures for Generative AI models for training and inference and showcase benefits on AMD

Work with open-source framework and community (e.g., PyTorch, JAX, Hugging Face) to integrate AMD optimized models, libraries and publish training recipes

Collaborate with software and hardware team to E2E co-optimize performance on current and future AMD solutions

Publish and promote your work within AMD and at external venues

PhD or master's degree or above; with major in CS, EE, Mathematics, or a related field

Preferred

Strong technical expertise in Gen AI model training and inference, and familiarity working with deep learning frameworks like PyTorch/JAX

Strong technical expertise in algorithmic innovation towards efficient Gen AI application for both training and inferencing

Expertise/publications in one of the areas preferred - efficient model architectures, optimized training, innovative parallelism strategies or low-precision training

Additional plus if publications include conferences such as NeuRIPS, CVPR, ECCV/ICCV, ICML, ICLR, etc

Experience productizing generative AI models and training foundation model at Scale

Excellent written, verbal, and presentation skills, ability to coordinate internally and externally

Several years of experience in AI, deep learning and related software development