Senior GPU Compiler Software Development Engineer

3 Months ago • 3-5 Years • Research & Development

Job Summary

Job Description

Senior GPU Compiler Software Development Engineer focusing on GPU support for OpenAI/Triton, a language and compiler for efficient custom deep-learning primitives. Responsibilities include new feature development, optimization, and support for Triton on GPUs with ROCm. This role requires collaboration with developers, customers, and project managers. Testing, documentation, and verification using unit/component/functional tests are crucial. The work involves analyzing, developing, and deploying performance improvements for neural networks implemented with Triton on GPUs.
Must have:
  • Strong C/C++ programming skills
  • GPGPU computing experience (HIP, CUDA, OpenCL)
  • Compiler internals experience (LLVM, GCC)
  • Basic Python programming skills
  • Performance analysis experience
Good to have:
  • Understanding of ML technologies
  • PyTorch experience
  • LLVM and MLIR compiler infrastructure experience
  • ROCm infrastructure knowledge
  • CMake, make/ninja build system experience
  • GEMM performance fundamentals
  • Docker experience

Job Details

Project description

Working on GPU support for OpenAI/Triton — a language and compiler for writing highly efficient custom Deep-Learning primitives. Work with the open-source community to analyze, develop, test, and deploy performance improvements for neural networks implemented with Triton on GPUs with ROCm.

Responsibilities

New features development, support and optimization of OpenAI/Triton project for GPUs. Communication with other developers, customers and project managers. Test implementation, project documentation and verification of system with unit/component/functional tests.

Skills

Must have

Strong C/C++ programming skills

Experience with GPGPU (General purpose GPU) computing (HIP, CUDA, OpenCL, etc.) or Experience with compiler internals (llvm, gcc or any other)

Basic Python programming skills

Experience in performance analysis

Nice to have

Basic understanding of ML technologies

Experience with PyTorch

Experience with LLVM and MLIR compiler infrastructure, analysis or optimizations implementation

Knowledge of ROCm infrastructure

Experience in CMake, make/ninja build system

GEMM performance fundamentals

Experience with Docker

Other

Languages

English: B2 Upper Intermediate

Seniority

Senior

Similar Jobs

Microsoft - Research Intern - Applied Sciences Group (Audio/Vision/NLP/Multimodal)

Microsoft

Redmond, Washington, United States (On-Site)
1 Month ago
ByteDance - Machine Learning Engineer Intern (Global E-commerce Risk Control) - 2025 Summer (MS)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Novancy One | Digital Talent Recruitment - Expert data scientists/Researcher in Generative AI Ref. 005529

Novancy One | Digital Talent Recruitment

Massachusetts, United States (On-Site)
5 Months ago
Tencent - Research Intern (NLP)

Tencent

Palo Alto, California, United States (On-Site)
2 Days ago
Google - Software Engineer, PhD, Early Career, Campus, AI/Machine Learning, 2025 Start

Google

Atlanta, Georgia, United States (On-Site)
3 Months ago
Mobileum - Architect- System Engineering

Mobileum

Bengaluru, Karnataka, India (Hybrid)
5 Months ago
NVIDIA - Senior Digital Circuit Design Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Tesla - Mechanical Design Engineer Internship

Tesla

Brandenburg, Germany (On-Site)
1 Week ago
Intel Corporation - Senior Formal Verification Engineer

Intel Corporation

Haifa District, Israel (Hybrid)
2 Months ago
Ubisoft - Research Student - Ubisoft La Forge

Ubisoft

Shanghai, Shanghai, China (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Salesforce - Principal Data Scientist

Salesforce

Palo Alto, California, United States (On-Site)
3 Months ago
Playrix - Feature Owner (LiveOps)

Playrix

Serbia (Remote)
4 Months ago
Microsoft - Senior Machine Learning Research Scientist

Microsoft

Cambridge, England, United Kingdom (On-Site)
1 Month ago
G5 Games - 2D HOG Grind Artist

G5 Games

Limassol, Limassol, Cyprus (Remote)
2 Weeks ago
Arkose Labs - Senior Machine Learning Researcher

Arkose Labs

Pune, Maharashtra, India (Hybrid)
4 Months ago
G5 Games - 2D HOG Grind Artist

G5 Games

Tbilisi, Tbilisi, Georgia (Remote)
2 Weeks ago
NVIDIA - Senior Deep Learning Software Engineer, cuDNN

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
NVIDIA - Senior Hypervisor and RTOS Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
NVIDIA - Senior AI Training Performance Engineer

NVIDIA

Shanghai, Shanghai, China (Hybrid)
1 Month ago
PwC - Conversational AI Developer- Manager

PwC

Bengaluru, Karnataka, India (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Brazil, Indiana, United States

Appier - Campaign Analyst

Appier

San Francisco, California, United States (On-Site)
2 Months ago
Mattel  Inc  - Senior Product Designer - Barbie

Mattel Inc

California, United States (On-Site)
2 Months ago
IGT - Director, Accounting- Controller

IGT

Reno, Nevada, United States (On-Site)
3 Months ago
The Walt Disney Company - Senior Software Engineer (Swift)

The Walt Disney Company

Seattle, Washington, United States (On-Site)
1 Month ago
BigID - Regional Sales Manager - Midwest (Wisconsin)

BigID

Madison, Wisconsin, United States (Remote)
3 Months ago
Riot Games - Principal Researcher

Riot Games

Los Angeles, California, United States (On-Site)
3 Months ago
The Walt Disney Company - Disney Live Entertainment Art Studio Intern, Summer/Fall 2025

The Walt Disney Company

Kissimmee, Florida, United States (On-Site)
2 Weeks ago
The Walt Disney Company - Royalties Analyst

The Walt Disney Company

Burbank, California, United States (On-Site)
5 Days ago
Zones - Field Services Technician

Zones

Texas, United States (On-Site)
2 Months ago
PENN Interactive - Motion Designer

PENN Interactive

Philadelphia, Pennsylvania, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Rivos - Silicon DFT - Full time

Rivos

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
ByteDance - Design Verification Engineer - Multimedia Lab

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Corsair - Hardware Development Engineer

Corsair

Vietnam (On-Site)
1 Month ago
ByteDance - Technical Expert, Large Language Model

ByteDance

Singapore (On-Site)
3 Months ago
NVIDIA - Senior Solutions Architect, CSP System

NVIDIA

Beijing, Beijing, China (On-Site)
1 Month ago
Rivos - Silicon Microarchitecture & Logic Design - Intern

Rivos

Santa Clara, California, United States (On-Site)
4 Months ago
Rivos - Silicon Logic Formal Verification - Full Time

Rivos

Portland, Oregon, United States (Hybrid)
4 Months ago
NVIDIA - Senior Software Aerial Performance Engineer

NVIDIA

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Rivos - Accelerator Design Verification - Full Time

Rivos

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
NVIDIA - Senior Hypervisor and RTOS Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

Luxoft, a DXC Technology Company (NYSE: DXC), is a digital strategy and software engineering firm providing bespoke technology solutions that drive business change for customers the world over. Acquired by U.S. company DXC Technology in 2019, Luxoft is a global operation in 44 cities and 21 countries with an international, agile workforce of nearly 18,000 people. It combines a unique blend of engineering excellence and deep industry expertise, helping over 425 global clients innovate in the areas of automotive, financial services, travel and hospitality, healthcare, life sciences, media and telecommunications.

DXC Technology is a leading Fortune 500 IT services company which helps global companies run their mission critical systems. Together, DXC and Luxoft offer a differentiated customer-value proposition for digital transformation by combining Luxoft’s front-end digital capabilities with DXC’s expertise in IT modernization and integration. Follow our profile for regular updates and insights into technology and business needs.

Gothenburg, Västra Götaland County, Sweden (On-Site)

New Delhi, Delhi, India (Remote)

Poland, Ohio, United States (Remote)

Kraków, Lesser Poland Voivodeship, Poland (On-Site)

Wrocław, Lower Silesian Voivodeship, Poland (On-Site)

Ukrainka, Kyiv Oblast, Ukraine (Remote)

Kuala Lumpur, Federal Territory Of Kuala Lumpur, Malaysia (On-Site)

Bengaluru, Karnataka, India (On-Site)

Mississauga, Ontario, Canada (On-Site)

View All Jobs

Get notified when new jobs are added by Luxoft

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug