21 hours ago
Santa Clara, CA, USAIntern
Responsibilities
- Develop and optimize infrastructure for distributed training, reinforcement learning, inference, and model optimization.
- Build scalable systems for experiment management, rollout generation, model serving, and evaluation.
- Improve training and inference efficiency through parallelism, scheduling, checkpointing, caching, and performance optimization.
- Develop tools integrating AI models with engineering systems, evaluation pipelines, and benchmarking frameworks.
- Support experiment tracking, logging, monitoring, and reproducibility across machine learning workflows.
- Analyze system performance and identify improvements in reliability, scalability, and resource utilization.
- Collaborate with researchers and engineers to translate experimental requirements into production-quality infrastructure.
- Document system designs, experimental results, and technical implementations.
Requirements
- Currently enrolled in a PhD program in Computer Science, Computer Engineering, Electrical Engineering, Artificial Intelligence, or a related field.
- Programming experience in Python.
- Coursework or hands-on experience in artificial intelligence or computer systems.
- Familiarity with PyTorch or similar machine learning frameworks.
- Interest in distributed systems, reinforcement learning, LLMs, or AI infrastructure.
- Strong analytical, problem-solving, and communication skills.
Benefits
- Full-time, 40-hour-per-week hybrid or onsite work in Santa Clara, California, throughout the internship.
- Summer 2027 semester internship dates are May 24, 2027 through August 13, 2027, or quarter internship dates are June 21, 2027 through September 10, 2027.
- AMD benefits are described in its benefits-at-a-glance materials.
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.