3 days ago
San Jose, CA, USASenior
Base Salary
$204k - $306k/yr
Responsibilities
- Integrate the compiler and compiled models with ML frameworks such as ONNX and PyTorch.
- Implement model partitioning in ML frameworks and/or MLIR.
- Develop runtime software to distribute work to and collect results from x86 and AIE cores.
- Design and develop new AMD technologies.
- Debug existing issues and research more efficient implementation approaches.
Requirements
- Bachelor's or master's degree in computer engineering, software engineering, computer science, or a related technical discipline.
- Strong object-oriented programming background with C/C++ and Python.
- Experience with ML compiler and runtime technologies such as MLIR and LLVM.
- Compiler-building skills and experience generating code for machine learning hardware accelerators.
- Knowledge of machine learning concepts, model development, ONNX, and PyTorch.
- Experience with modern concurrent programming and threading APIs is preferred.
- Experience with debuggers, source-code control systems such as GitHub, and profilers is a plus.
- Strong communication, problem-solving, code-quality, and attention-to-detail skills.
Categories
About AMD
AMD designs and sells CPUs, GPUs, and adaptive/embedded computing products for PCs, data centers, gaming, and edge devices. Its portfolio includes Ryzen and EPYC processors, Radeon and Instinct graphics, and adaptive SoCs from its Xilinx acquisition, sold to OEMs, cloud providers, and device makers. Founded in 1969 and headquartered in Santa Clara, it is a public company on NASDAQ and supplies semi-custom chips for major game consoles.