onsite
Software Engineer - AI Models - AMD
Software Engineer
Lead the design and deployment of cutting‑edge AI models, leveraging Python, PyTorch, and GPU programming to accelerate next‑generation computing solutions.
About the role
Key Responsibilities
- Design, implement, and optimize AI models for high‑performance computing environments.
- Collaborate with cross‑functional teams to integrate models into AMD’s product portfolio.
- Develop scalable training pipelines using PyTorch and GPU‑accelerated frameworks.
- Conduct rigorous performance evaluation and fine‑tune models for real‑world workloads.
- Document architecture, code, and best practices for internal knowledge sharing.
Requirements
- Strong proficiency in Python and experience with deep learning libraries such as PyTorch.
- Deep understanding of machine learning concepts and model optimization techniques.
- Hands‑on experience with GPU programming and performance profiling.
- Excellent problem‑solving skills and ability to work in a fast‑paced, collaborative environment.
- BS/MS in Computer Science, Electrical Engineering, or related field.
Skills
pythonmachine learningdeep learningpytorch