Be an early applicantLess than 25 applicants

This job has closed.

Company

Original Job Post

AMD · 2 days ago

SMTS Software System Design Engineer

Austin, TX

Full-time

Hybrid

Senior Level

Wonder how qualified you are to the job?

Maximize your interview chances

Cloud ComputingComputer

H1B Sponsorship

Growth Opportunities

Insider Connection @AMD

Discover valuable connections within the company who might provide insights and potential referrals, giving your job application an inside edge.

Responsibilities

Design and implement NPU compiler framework for neural networks.

Develop hardware aware graph optimizations for high level ML frameworks like ONNX.

Research new algorithms for operator scheduling for efficient inference of latest NN models.

Interface with ONNX / Pytorch runtime and lower level HW implementation.

Contribute to high performance inference for GenAI workloads such as Llama2-7B, Stable diffusion, SDXL-Turbo etc.

Work closely with kernel developers, performance architects, and AI researchers

Manage CPU, and memory resources effectively during model execution.

Handle resource allocation for ML deployments across different tenants.

Research heterogenous mapping of ML operators for maximum efficiency.

Build tools to track resource utilization, bottlenecks, and anomalies.

Enable detailed profiling and debugging tools for analyzing ML workload latency.

Implement rigorous code review practices for superior code quality assurance.

Qualification

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

C++PythonTVMMLIRONNXPyTorchCNNLSTMLLMsDiffusionRuntime IntegrationProblem-SolvingPassion

Required

Strong programming skills in C++, Python.

Experience with proprietary/open source compiler stack: TVM, MLIR.

Experience with ML frameworks (e.g., ONNX, PyTorch) is required.

Experience with ML models such as CNN, LSTM, LLMs, Diffusion is a must.

Experience with ONNX, Pytorch runtime integration is a bonus.

Excellent problem-solving abilities and a passion for performance optimization.

Master’s, or PhD degree in Computer Science, Electrical Engineering, or related fields.

Benefits

Annual bonus or sales incentive

Employee Stock Purchase Plan

Competitive benefits

Company

AMD

Glassdoor

4.1

Advanced Micro Devices (AMD) is a semiconductor company that designs and develops graphics units, processors, and media solutions.

Founded in 1969

Santa Clara, California, USA

10,001+ employees

http://www.amd.com

H1B Sponsorship

AMD has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Trends of Total Sponsorships

2023 (465)

2022 (689)

2021 (578)

2020 (559)