64 applicants

Company

Original Job Post

AMD · 2 days ago

AI Software Development Engineer

Austin, TX

Full-time

Onsite

Mid 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

Work with a team of Software Engineers to enable DL models, libraries, and applications for Instinct GPUs in both on-prem and Cloud environments.

Analyze and optimize the performance of AI software and understand hardware bottlenecks to improve performance.

Collaborate with community maintainers and submit contributions for open-source software development.

Implement and optimize parallel methods on GPU accelerators and conduct performance analysis for both CPU and GPU.

Provide clear and timely communication related to project status and key aspects to the leadership team.

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++PythonDeep LearningPytorchTensorflowOpen-Source DevelopmentPerformance OptimizationSoftware DevelopmentQuality AssuranceProblem-SolvingAnalyticalTeamworkWillingness to LearnProfiling ToolsAI SW StackGPU AcceleratorsParallel MethodsPerformance AnalysisSingularityDockerKubernetesClear CommunicationTimely Communication

Required

Strong programming skills in C++ and/or Python

Strong development experience is at least one major DL framework such as Pytorch or Tensorflow in inference, fine-tuning and/or training

BS/MS with years of related experience or PhD with years of related experience in Computer Science or Computer Engineering or related equivalent

Experience with open-source software development including collaboration with community maintainers and submitting contributions is a plus

Excellent analytical and problem-solving skills root-causing/addressing performance issues

Ability to work independently and as part of a team

Willingness to learn skills, tools, and methods to advance the quality, consistency, and timeliness of AMD software products

Preferred

Expertise in profiling tools across the AI SW Stack (Torchprofiler, RocM profiler, Vtune, Nsight)

Experience in implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI)

Performance analysis skills for both CPU and GPU

Experience with Singularity, Docker, and/or Kubernetes

Experience providing clear and timely communication related to status and other key aspects of the project to leadership team

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)