JobsPrincipal AI Software Engineer
Job description
AMD AI Group is seeking a technical leader for OneROCm, focusing on driving a unified ROCm software stack across AMD's diverse product portfolio. This role presents a unique opportunity to influence innovation and develop next-generation products at a company-wide scale. The ideal candidate will shape the end-to-end ROCm software, requiring strong hardware/software co-design to maximize performance across various AMD products. The position emphasizes collaboration, technical excellence, and the ability to engage with top customers to meet their unique workload requirements.
Requirements
- Deep expertise in at least one major area of the software stack.
- Knowledge in GPU architectures and basic knowledge of CPU architecture.
- Experience in AI/ML software stack spanning compilers, kernels, runtime, libraries, models, frameworks, and performance optimization layers.
- Understanding of GPU programming such as ROCm, CUDA, and OpenCL.
- Experience in hardware/software co-design and building high-performance products across the full product lifecycle.
- Undergraduate degree required, with a preference for a Bachelor of Science, Masters, or PhD in Electrical Engineering, Computer Architecture, or Computer Science.
Responsibilities
- Collaborate across hardware architecture, compiler, math libraries, kernel, and framework teams to influence future silicon features based on evolving AI workload trends.
- Deliver innovations and roadmap for the AI software stack across all AMD products.
- Lead the profiling, analysis, and tuning of large-scale models to ensure performance excellence on AMD hardware.
- Drive the development of advanced tools and frameworks for performance estimation, modeling, and automated reporting.
- Partner with top customers and hyperscalers to understand their unique workload requirements and deliver tailored architectural wins and software optimizations.
- Mentor and inspire other engineers and contribute to ROCm Open Source.
Benefits
- AMD provides a competitive 'Total Rewards' package that focuses on financial growth, health, and work-life balance.
Is this posting expired or inaccurate?
