Fall 2024 PhD AI and ML Co-Op/Intern

Key Skills

AI technology, Algorithms, Apache TVM, C Programming, C++, Conda, CUDA kernels, Deep Learning, Docker, Image processing, Linux Operating system, Machine learning techniques, Neural Networks, ONNX, Optimization, Python Programming, PyTorch, Rust, TensorFlow

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_

As a Co-op student, you can make an immediate contribution to AMD's next generation of technology innovations. We have a dynamic, high-energy work environment, filled with expert employees, and unique opportunities for developing your career. You will have the opportunity to connect with AMD leaders, receive one-on-one mentorship, attend amazing networking events and much more. With AMD, you can get hands-on experience that will give you a competitive edge in the workforce.

Get a jumpstart on an exciting career - apply today!

LOCATION: Austin, TX

ONSITE: This role requires the student to work from the Austin office throughout the duration of the Co-op/Intern term.

PROGRAM TERM:

• Fall 2024: Start Date August 19th or September 9th, 2024

RESPONSIBILITIES:

• You will be responsible for executing projects for AI software framework development and benchmarking.
• You will be involved in model migration to our RyzenAI stack and ensuring that the models are quantized and optimized to run on the AI stack.
• You will research on model inference acceleration and involved in software simulation.
• You will be participating in research and development of deployment solutions for deep learning models on embedded devices.
• You will be participating in the development of AI compilers and related tools
• You will work very closely with dev teams and Project Managers to drive results.
• You will be writing technical documents and reports, participating in technical exchanges and presentations.

KEY QUALIFICATIONS:

• You are currently enrolled in an accredited US University into a Masters or PhD degree program majoring in Computer Engineering, Electrical Engineer, Computer Science or related discipline with a focus on AI/ML.
• You have good programming ability in Python and/or C++. Knowledge of C will be a bonus.
• You are familiar with some popular development tools, e.g. git, cmake, make, conda or docker
• You have basic knowledge about data structure and algorithms, algorithm optimization, neural network, computer architecture.
• You have proficiency in common deep learning frameworks such as PyTorch, TensorFlow etc.

PREFERRED QUALIFICATIONS:

• You have familiarity with Rust Programming.
• You have experience on AI framework development or operator implementation.
• You have knowledge of ONNX.
• You have a basic knowledge about data structure and algorithm, algorithm optimization, image processing, neural network, computer architecture.
• You have some experience with ML quantization and AI compilers and familiarity with existing framework like TVM, LLVM, MLIR.
• You have experience in driver development and Linux kernel/driver development

BENEFITS:

• Healthcare coverage, dental and vision
• Paid holidays
• Relocation stipend
• Education assistance for required Co-op/Intern course.