3 days ago
San Jose, CA, USASenior
Base Salary
$240k - $360k/yr
Responsibilities
- Develop, train, fine-tune, and optimize generative AI models for image generation and visual content creation.
- Design and implement diffusion models, multimodal architectures, and advanced machine learning algorithms.
- Build scalable AI training and inference pipelines using high-performance GPU computing.
- Evaluate model quality, performance, and efficiency with advanced techniques and benchmarks.
- Collaborate with research, software, and hardware teams to integrate AI innovations into AMD technologies.
- Optimize AI workloads for AMD GPU architectures and contribute to performance improvements.
- Track advancements in generative AI, image synthesis, diffusion models, and large-scale machine learning.
Requirements
- Strong software development experience in Python and C++.
- Deep understanding of machine learning, deep learning, and neural network architectures.
- Experience developing and training diffusion models, image generation models, GANs, transformers, or multimodal AI systems.
- Hands-on experience with PyTorch, TensorFlow, JAX, or similar machine learning frameworks.
- Experience fine-tuning and deploying large-scale AI models.
- Familiarity with distributed training, model optimization, and GPU acceleration techniques.
- Experience building production AI training and inference pipelines.
- Knowledge of computer vision, image synthesis, and multimodal learning techniques.
- Experience with MLOps, CI/CD workflows, Git, and modern software engineering practices.
- Experience with AI coding assistants, agentic AI systems, or automated development workflows.
- Strong communication, collaboration, and problem-solving skills.
- A master's degree or PhD in Computer Science, Artificial Intelligence, Machine Learning, Computer Engineering, or a related technical field is preferred.
Categories
About AMD
AMD designs and sells CPUs, GPUs, and adaptive/embedded computing products for PCs, data centers, gaming, and edge devices. Its portfolio includes Ryzen and EPYC processors, Radeon and Instinct graphics, and adaptive SoCs from its Xilinx acquisition, sold to OEMs, cloud providers, and device makers. Founded in 1969 and headquartered in Santa Clara, it is a public company on NASDAQ and supplies semi-custom chips for major game consoles.