AI Optimization Engineer

AMD
Shanghai
Workplace: OnsiteFull timeCNY 526,260 - 751,800 annuallyFunction: Data Science & Machine LearningEducation: bachelorsSkills: ["Leadership","Technical problem-solving","Collaboration","Communication"]

Build and optimize high-performance software and features for next-generation AI and product-introduction projects. Support key customer engagements with fast, robust technical help, develop optimization features and validate them before release, and collaborate with dev teams and project managers to deliver scalable, durable systems. Leverage AI agent tools for testing and analysis, and apply deep expertise in C/C++, Python, CUDA/Triton kernels, and performance profiling to drive benchmark and efficiency improvements.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
AMD
AMD
18 hours ago

AI Optimization Engineer

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 12 hours agoStatus: Live

Job Summary

Build and optimize high-performance software and features for next-generation AI and product-introduction projects. Support key customer engagements with fast, robust technical help, develop optimization features and validate them before release, and collaborate with dev teams and project managers to deliver scalable, durable systems. Leverage AI agent tools for testing and analysis, and apply deep expertise in C/C++, Python, CUDA/Triton kernels, and performance profiling to drive benchmark and efficiency improvements.
Location: Shanghai
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning

Key Responsibilities

  • •Develop and drive execution of comprehensive, highly effective software for sophisticated technology and new product introduction projects.
  • •Provide quick and robust technical support for key customer projects.
  • •Develop optimization features and validate them before releasing to customers.
  • •Contribute to a high-functioning feature team and collaborate with multiple teams to deliver key planning solutions.
  • •Collaborate on the design and implementation of future architecture for highly scalable, durable systems while working closely with dev teams and project managers.

Pay and Benefits

Salary: CNY 526,260 - 751,800 annually

Key Requirements

  • •Expert knowledge and hands-on experience in C, C++, and Python.
  • •Familiar with AI operations libraries such as flashInfer, tensorRT-LLM, and CUTLASS, including using/customizing ops and ops fusion.
  • •Experience with Triton compiler backend, including llir/ASM with explicit layout and pipeline optimization.
  • •Ability to implement CUDA/Triton/Gluon kernels and reach at least 50% MFU with compute-dense kernels.
  • •Bachelor’s or Master’s degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent.
Experience:AIPerformance optimizationGPU kernelsCompiler backends
Education:Bachelor's in Computer Science; Computer Engineering; Electrical Engineering
Skills:LeadershipTechnical problem-solvingCollaborationCommunication
Languages:En-us
Tech Stack:CC++PythonFlashInferTensorRT-LLMCUTLASSAiter(AMD)TritonLlirASMCUDAGluonCUDA/Triton/Gluon kernelsNsightROCprofilerAMD CNDARDNASIMD

Company Brief

AMD
Designs and produces semiconductor products including CPUs, GPUs, and adaptive SoCs for consumer, enterprise, and embedded markets, competing across PCs, data centers, and gaming industries.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1969
Glassdoor
Glassdoor: 3.9
WebsiteLinkedIn