Senior GPU Compiler Software Development Engineer

3 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, implementing tests, and documenting the project. The work involves improving the performance of neural networks on GPUs using ROCm. The ideal candidate possesses strong C/C++ skills, GPGPU experience (HIP, CUDA, OpenCL), and compiler internals knowledge (LLVM, GCC).
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

Zoox - Collision Avoidance System, Machine Learning Internship/Co-op

Zoox

Foster City, California, United States (On-Site)
4 Months ago
NVIDIA - Deep Learning Solution Architect

NVIDIA

Shanghai, Shanghai, China (On-Site)
1 Month ago
Playrix - Feature Owner (LiveOps)

Playrix

Ukraine (Remote)
4 Months ago
Mistplay - Staff Machine Learning Engineer I (MLE)

Mistplay

Montreal, Quebec, Canada (Hybrid)
2 Weeks ago
Passive Logic - AI Control Theory & Optimization Scientist

Passive Logic

Salt Lake City, Utah, United States (On-Site)
2 Months ago
Riot Games - Researcher III - Teamfight Tactics

Riot Games

Dublin, County Dublin, Ireland (On-Site)
3 Months ago
bosh group india - Circuit Analysis Engineer - Team Lead

bosh group india

Bengaluru, Karnataka, India (On-Site)
2 Weeks ago
Nagarro - Senior Staff Engineer, Machine Learning

Nagarro

Bengaluru, Karnataka, India (On-Site)
1 Month ago
ByteDance - Senior Machine Learning Ops Engineer, ML System - Foundation Model

ByteDance

Seattle, Washington, United States (On-Site)
2 Weeks ago
JMA - Senior Engineer - Firmware

JMA

Bologna, Emilia-Romagna, Italy (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

NVIDIA - Senior AI Training Performance Engineer

NVIDIA

Shanghai, Shanghai, China (Hybrid)
1 Month ago
Trendyol - Data Science Team Lead - Dolap

Trendyol

İstanbul, İstanbul, Türkiye (Hybrid)
1 Month ago
LeoVegas - Data Scientist - Sportsbook

LeoVegas

Stockholm, Stockholm County, Sweden (Hybrid)
2 Months 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
Hitachi - Artificial Intelligence - JBU

Hitachi

Chennai, Tamil Nadu, India (On-Site)
4 Months ago
PwC - Conversational AI Developer- Senior Associate

PwC

Bengaluru, Karnataka, India (On-Site)
4 Months ago
Playrix - Feature Owner (LiveOps)

Playrix

Ukraine (Remote)
4 Months ago
G5 Games - 2D UI/UX Artist (Hidden objects project)

G5 Games

Tbilisi, Tbilisi, Georgia (Remote)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in undefined

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Research & Development Jobs

Riot Games - Senior Manager, Technical Product Management - VALORANT, Foundations

Riot Games

Los Angeles, California, United States (On-Site)
1 Month ago
Anavation - Lead Software Engineer

Anavation

Chantilly, Virginia, United States (On-Site)
4 Months ago
Microsoft - Research Intern - Knowledge and Reasoning

Microsoft

Redmond, Washington, United States (On-Site)
1 Month ago
Luxoft - Embedded Linux Development Engineer

Luxoft

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Luxoft - Senior GPU Kernel Developer

Luxoft

Italy, New York, United States (Remote)
2 Months ago
ByteDance - Backend Engineer, Machine Learning Systems - Singapore

ByteDance

Singapore (On-Site)
3 Months ago
The Walt Disney Company - Senior Generalist

The Walt Disney Company

Sydney, New South Wales, Australia (On-Site)
5 Months ago
Fabric - Applied Researcher, Cryptography Proof Systems

Fabric

New York, New York, United States (Remote)
4 Months ago
ByteDance - Research Scientist Graduate (High-Performance Computing (Inference Optimization) - Vision AI Platform)

ByteDance

San Jose, California, United States (On-Site)
5 Days ago
NVIDIA - Senior Memory Controller Verification Engineer

NVIDIA

Toronto, Ontario, Canada (On-Site)
3 Weeks 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