Principal Deep Learning Data Scientist, AI Solutions
Job location
Santa Clara, California, US
Type
Full Time
Responsibilities:
Understanding and prototyping customer AI workloads and working with the optimization team on accelerating models for AMD infrastructure.
Delivering AI solutions in line with product roadmap on time with high quality.
Involved in cross-function team interaction and engagement with various key players and stakeholders in the organization.
Play a role in all the phases of software & model development, from requirement gathering, analysis, design, development, testing, and final release to customers.
Provide clear and timely communication related to status and other key aspects of the project to the leadership team.
Responsible for working with customers and internal support teams to resolve critical customer escalations on a need basis.
Willingness to learn skills, tools, and methods to advance the quality, consistency, and timeliness of AMD software products.
Experience with deep learning models and associated AI frameworks (PyTorch, TensorFlow, Jax (FLAN))
Requirements & Skills:
Deep experience with AI/ML frameworks like Pytorch, Tensor Flow, JAX, etc. is preferred.
Knowledge of Acceleration platforms like GPU, TPU, APU, and FPGAs in preferred
BS with several years of related experience MS with years of related experience or PhD with years of related experience in Computer Engineering or Electrical Engineering or Computer Science or related equivalent from top universities
