Principal ML/AI Solutions Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Advanced Microdevices Pvt. Ltd. (India) · 9 hours ago

Principal ML/AI Solutions Engineer

Advanced Micro Devices, Inc. is focused on building innovative products that enhance computing experiences, including AI and data centers. They are seeking a Principal ML/AI Solutions Engineer to empower customers in adopting AMD’s AI software stack, requiring strong technical skills in machine learning and excellent communication abilities.

BiopharmaBiotechnologyIndustrialManufacturing
badNo H1Bnote

Responsibilities

Work across reference design creation, workflow optimization, performance tuning, and ecosystem enablement
Collaborate with engineering teams, ISVs, cloud partners, and enterprise customers will be essential to guide real-world AI deployments and translate customer insights into actionable engineering direction
Drive innovation by prototyping and demonstrating end-to-end AI workflows across training, fine-tuning, and inference; developing validated designs, notebooks, and performance playbooks; and providing structured feedback to AMD’s ROCm, frameworks, platform, and performance engineering teams
Champion AMD technologies within the AI community, build demos and best-practice content, influence product roadmaps, validate ecosystem integrations such as vLLM and Triton, and support partner technology readiness—ultimately advancing AMD’s AI ecosystem and improving workload efficiency at scale

Qualification

ML/AI engineeringGPU computingCustomer engagementPyTorchTensorFlowDistributed trainingEnterprise solution designTechnical advocacyInnovationCommunication skillsProblem-solvingCollaboration

Required

Strong technical depth in machine learning frameworks, GPU workflows, and enterprise architecture
Excellent communication and customer-facing skills
Deep expertise in ML/AI systems
Proven track record in collaboration, problem-solving, and technical execution
Ability to engage directly with enterprise stakeholders to understand complex workloads
Experience leading technical deep dives, delivering reference architectures, and guiding solution integrations
Experience in reference design creation, workflow optimization, performance tuning, and ecosystem enablement
Ability to collaborate with engineering teams, ISVs, cloud partners, and enterprise customers
Drive innovation by prototyping and demonstrating end-to-end AI workflows across training, fine-tuning, and inference
Develop validated designs, notebooks, and performance playbooks
Provide structured feedback to AMD's ROCm, frameworks, platform, and performance engineering teams
Champion AMD technologies within the AI community
Build demos and best-practice content
Influence product roadmaps and validate ecosystem integrations
Support partner technology readiness
BS, MS, or PhD in Computer Science or a related equivalent

Preferred

Extensive expertise in ML/AI engineering, distributed training, and GPU computing
Strong experience in customer engagement, enterprise solution design, and technical advocacy
Knowledge of modern ML workloads (LLMs, generative AI, RAG, CV, fine-tuning)
Hands-on experience with PyTorch, TensorFlow, JAX, inference runtimes, and containerized GPU environments
Familiarity with ROCm, CUDA, AMD MI300/MI325 platforms, MLOps systems, data pipelines, and cloud-native deployment architectures
Additional experience developing demos, validated designs, reproducible notebooks, or technical content
Strong understanding of industry trends and competitive AI technologies

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