Position Details

Type Not Specified

Experience senior

Exp. Years Not specified

Education Not specified

Category AI & Machine Learning

About this role

This role involves developing and optimizing AI models on AMD GPUs, enhancing frameworks like PyTorch, and contributing to open-source AI tools to support high-performance AI workloads.

Key Responsibilities

Model development and optimization
Framework enablement
Open-source integration
Community collaboration
System performance tuning

Technical Overview

Focuses on model development, performance optimization, framework enablement, and open-source collaboration within GPU-accelerated AI environments.

Ideal Candidate

The ideal candidate is a senior AI software engineer with expertise in C++, Python, and deep learning frameworks like PyTorch. They have a strong background in GPU performance optimization and open-source contributions.

Must-Have Skills

C++PythonAI developmentDeep LearningPyTorchGPU performance optimization

Nice-to-Have Skills

Open-Source contributionsDistributed systemsModel optimization

Tools & Platforms

PyTorchJAXTritonROCmATen

Required Skills

C++PythonAI developmentDeep LearningModel OptimizationPyTorchGPU performance tuningOpen-Source contributions

Hard Skills

C++PythonAI developmentDeep LearningModel OptimizationPyTorchOpen-SourceGPUPerformance tuningDistributed training

Soft Skills

CollaborationAnalytical skillsIndependenceGoal settingProblem-solving

Industry & Role

Industry Semiconductor / Hardware

Job Function AI model optimization and framework support for AMD GPUs

Role Subtype AI Software Engineer

Tech Domains Python, C++, PyTorch, JAX, Triton, ROCm

Keywords for Your Resume

PyTorch Software Development EngineerC++PythonAI developmentDeep LearningModel OptimizationFramework EnablementOpen-SourceGPUPerformance tuningDistributed trainingOpen-Source contributionsPyTorchGPU performanceModel optimizationDistributed systems

Deal Breakers

Lack of experience in AI development, No experience with PyTorch or deep learning frameworks, No GPU performance tuning skills

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile