Senior GPU Compiler Software Development Engineer

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

Job Summary

Job Description

This Senior GPU Compiler Software Development Engineer role focuses on enhancing OpenAI/Triton's GPU support. Responsibilities include developing new features, optimizing existing ones, collaborating with developers and customers, and conducting thorough testing using unit/component/functional tests. The work involves improving the performance of neural networks implemented with Triton on GPUs using ROCm. This position requires strong C/C++ skills, GPGPU computing experience (HIP, CUDA, OpenCL), and familiarity with compiler internals (LLVM, GCC). Performance analysis experience is a plus.
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
  • Experience with PyTorch
  • 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

Playrix - Game Designer

Playrix

Montenegro (Remote)
4 Months ago
ByteDance - Software Engineer, Inference

ByteDance

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

Playrix

Ukraine (Remote)
4 Months ago
Luxoft - Senior ML Engineer

Luxoft

Poland, Ohio, United States (Remote)
1 Month ago
ByteDance - Lead Research Scientist, Foundation Model, Speech & Audio

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
NVIDIA - Senior Verification Engineer

NVIDIA

Hsinchu, Hsinchu City, Taiwan (On-Site)
1 Month ago
Riot Games - Associate Art Director (Characters) - Unpublished R&D Product

Riot Games

Los Angeles, California, United States (On-Site)
4 Months ago
Rivos - Silicon Microarchitecture & Logic Design - Intern

Rivos

Santa Clara, California, United States (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Paypal - Staff Machine Learning Engineer

Paypal

San Jose, California, United States (Hybrid)
3 Months ago
NXP - Junior Developer of Systems Testing Infrastructure

NXP

Brno, South Moravian Region, Czechia (On-Site)
4 Months ago
NVIDIA - Senior Hypervisor and RTOS Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Recro - Automatic speech Recognition

Recro

Gurugram, Haryana, India (On-Site)
4 Months ago
NVIDIA - Senior AI Training Performance Engineer

NVIDIA

Shanghai, Shanghai, China (Hybrid)
1 Month ago
Google - Software Engineer, PhD, Early Career, Campus, AI/Machine Learning, 2025 Start

Google

Mountain View, California, United States (On-Site)
3 Months ago
Nintendo - Senior Data Scientist

Nintendo

Redmond, Washington, United States (On-Site)
1 Month ago
NVIDIA - Senior ASIC Verification Engineer, Coherent High Speed Interconnect

NVIDIA

Taipei City, Taiwan (On-Site)
1 Month ago
Modulate - Senior Machine Learning Engineer

Modulate

Somerville, Massachusetts, United States (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Italy, New York, United States

Whoop - IT Manager

Whoop

Boston, Massachusetts, United States (On-Site)
4 Months ago
Google - Staff Software Engineer, Google Cloud Compute

Google

Seattle, Washington, United States (On-Site)
3 Months ago
Corsair - Influencer Gaming Content Intern

Corsair

Duluth, Georgia, United States (On-Site)
4 Days ago
Fandom - Staff Fullstack Software Engineer (Javascript) - Ad Tech

Fandom

San Francisco, California, United States (Remote)
3 Months ago
Fluence - Chief Battery Energy Storage System (BESS) Engineer

Fluence

Arlington, Virginia, United States (Hybrid)
4 Months ago
Luxoft - Calypso Techno-Functional Consultant

Luxoft

New York, New York, United States (On-Site)
3 Months ago
IGN - Senior Full Stack Software Engineer

IGN

Los Angeles, California, United States (Hybrid)
3 Months ago
Nagarro - Senior Staff Consultant, Business Analyst

Nagarro

New York, New York, United States (On-Site)
4 Months ago
ByteDance - Research Scientist- Foundation Model, Video Generation

ByteDance

Seattle, Washington, United States (On-Site)
3 Months ago
Microsoft - Research Intern - Algorithms Group: Reasoning Abilities of LLMs

Microsoft

Redmond, Washington, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

NVIDIA - Senior Analog Mixed Signal Design Engineer

NVIDIA

Ra'anana, Center District, Israel (On-Site)
1 Month ago
NVIDIA - Senior Physical Design Engineer

NVIDIA

Hsinchu, Hsinchu City, Taiwan (On-Site)
1 Month ago
Riot Games - Manager, Data Science - League Studio, League Data Central

Riot Games

Los Angeles, California, United States (On-Site)
3 Months ago
NVIDIA - Design Verification Intern - 2025

NVIDIA

Shenzhen, Guangdong Province, China (On-Site)
1 Day ago
Microsoft - Research Intern - Model Inference

Microsoft

Redmond, Washington, United States (On-Site)
1 Month ago
ByteDance - Research Engineer Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (PhD)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
NVIDIA - Firmware PHY Verification Engineer

NVIDIA

Beijing, Beijing, China (On-Site)
1 Month ago
Google - Senior Software Engineering Manager, Google Cloud

Google

Hyderabad, Telangana, India (On-Site)
1 Month ago
Microsoft - Senior Silicon Architect

Microsoft

Noida, Uttar Pradesh, India (On-Site)
1 Month ago
Assystems - Ingénieur coordination technique H/F

Assystems

Cherbourg-en-Cotentin, Normandy, France (On-Site)
3 Months 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