At AMD, the business problem is to accelerate next-generation computing experiences, from AI and data centers, to PCs, gaming and embedded systems, by building great products and driving end-to-end delivery of software solutions for inference and training of machine learning models.
Requirements
- Relevant experience in Machine Learning and/or GPU programming
- Experience in deep learning frameworks (e.g. TensorFlow, Keras, PyTorch, Caffe, ONNX, etc) and familiarity with CNN/LSTM model architectures
- Knowledge of CPU and GPU architecture, and experience in GPGPU programming technologies
Responsibilities
- Work within and coordinate with a small team to analyze, implement, and optimize DirectML-TensorFlow and PyTorch for machine learning models
- Collaborate with ISV, library, compiler, driver, and hardware engineers to influence strategic decisions to achieve the highest performance for DirectML
- Innovate new algorithmic improvements that exploit the strengths of the hardware architecture to deliver the best possible machine learning performance
Other
- Bachelor’s or Master’s degree in related discipline preferred
- Excellent communication and collaboration skills
- Experience advocating for technical solutions in a collaborative team environment
