AI Models Software Development Engineer

Overview In this role you will advance AI model development and optimization at AMD, focusing on NLP/Generative AI efficiencies. You will collaborate with software engineers, data scientists, and researchers to integrate models into platforms and scale their performance. You will push the boundaries of AI techniques, contribute to research publications, and stay at the forefront of algorithmic innovation. This position offers access to cutting-edge technology, mentorship, and a clear path for professional growth. Join us to help shape scalable AI solutions across industries. Compensation / Benefitscutting-edge technologytraining programsmentorship opportunitiesclear career progression path ResponsibilitiesResearch, design, and implement novel methods for efficient Generative AI training and inferenceAlgorithmic model optimization and training methods design including quantization, sparsity, and parallelism strategiesInnovate and publish research findingsCollaborate with software engineers, data scientists, and researchers to integrate AI models into software applications and platforms Key requirementsExperience with generative tasksFamiliar with deep learning frameworks (PyTorch/JAX/ONNX/TensorFlow)Experience with linear scaling of large language models and training at scaleExperience in model compression, quantization, and end-to-end inference optimizationStrong knowledge of AI, deep learning, or machine learningStrong Python coding skills; C/C++ a plusAcademic paper writing and technical innovationeffective communicationcollaborationcontinuous learningPyTorchJAXONNX