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 using ROCm. Collaboration with developers, customers, and project managers is crucial. The role requires testing, documentation, and verification using unit/component/functional tests. The position involves working with the open-source community to enhance neural network performance.
Must have:
  • Strong C/C++ programming skills
  • GPGPU experience (HIP, CUDA, OpenCL)
  • Compiler internals experience (LLVM, GCC)
  • Basic Python programming skills
  • Performance analysis experience
Good to have:
  • Basic 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

Paypal - Senior Machine Learning Engineer

Paypal

San Jose, California, United States (Hybrid)
3 Months ago
ByteDance - Senior AI Space Interaction Industrial Designer

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
ION - Senior AI Engineer, Italy

ION

Pisa, Tuscany, Italy (On-Site)
4 Months ago
NVIDIA - Manager, Tools and Development

NVIDIA

Pune, Maharashtra, India (On-Site)
1 Month ago
Visa - Principle Data Scientist, Visa Rules, Compliance and Standards (RCS)

Visa

Austin, Texas, United States (Hybrid)
2 Months ago
NVIDIA - Software Manager - BlueField DPU Platforms

NVIDIA

Massachusetts, United States (On-Site)
1 Month ago
Tencent - Game Research & Development Intern, Engine Research

Tencent

Palo Alto, California, United States (On-Site)
2 Months ago
Samsung Semiconductor - Principal Engineer, NPU Architect

Samsung Semiconductor

San Jose, California, United States (Hybrid)
4 Weeks ago
NVIDIA - Automation Software Engineer

NVIDIA

Yokne'am Illit, North District, Israel (On-Site)
3 Weeks ago
Meta - Software Engineer, Computer Vision (Technical Leadership)

Meta

Burlingame, California, United States (Remote)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

PwC - Conversational AI Developer- Senior Associate

PwC

Bengaluru, Karnataka, India (On-Site)
4 Months ago
Nintendo - Senior Data Scientist

Nintendo

Redmond, Washington, United States (On-Site)
1 Month ago
G5 Games - 2D HOG Grind Artist

G5 Games

Tbilisi, Tbilisi, Georgia (Remote)
2 Weeks ago
G5 Games - 2D UI/UX Artist (Hidden objects project)

G5 Games

Astana, Astana, Kazakhstan (Remote)
1 Month ago
Microsoft - AI Engineering: Internship Opportunities

Microsoft

Satakunta, Finland (On-Site)
1 Month ago
Zoox - Collision Avoidance System, Machine Learning Internship/Co-op

Zoox

Foster City, California, United States (On-Site)
4 Months ago
NVIDIA - Principal Engineer

NVIDIA

United States (Remote)
5 Days ago
NVIDIA - Senior System Software Engineer - QNX BSP and IO Virtualization

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Paypal - Staff Machine Learning Engineer

Paypal

San Jose, California, United States (Hybrid)
3 Months ago
Microsoft - Research Intern - Applied Sciences Group (Computer Vision)

Microsoft

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

Get notifed when new similar jobs are uploaded

Jobs in Poland, Ohio, United States

NVIDIA - Software Engineering Manager, Distributed Task-based Runtimes

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Blazesoft - Online Casino Program Manager

Blazesoft

Concord, California, United States (On-Site)
7 Months ago
Epic Games - Senior Software Engineer

Epic Games

United States (On-Site)
1 Month ago
Next Level Business Services - Performance Test Manager

Next Level Business Services

Babylon, New York, United States (On-Site)
4 Months ago
WebFX - Jr. Inside Sales Strategist

WebFX

Harrisburg, Pennsylvania, United States (On-Site)
4 Months ago
Take-Two Interactive - Global Payroll Tax Manager

Take-Two Interactive

New York, New York, United States (On-Site)
4 Months ago
Electronic Arts - Senior Manager, Anti-Cheat Engineering

Electronic Arts

Austin, Texas, United States (On-Site)
1 Month ago
Axiom Zen - Game Designer, CryptoKitties

Axiom Zen

United States (Remote)
2 Months ago
Salesforce - Sales Development Representative, Nonprofit (New Grad)

Salesforce

Illinois, United States (On-Site)
1 Month ago
Nagarro - Senior Staff Consultant, Business Analyst

Nagarro

New York, New York, United States (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

NVIDIA - Memory System Quality Architect

NVIDIA

California, United States (Hybrid)
1 Month ago
Google - Software Engineer (For Women in Tech Candidates)

Google

Belo Horizonte, State Of Minas Gerais, Brazil (On-Site)
3 Months ago
ByteDance - Software Engineer in Large Model System Graduate (Machine Learning Sys-US) - 2024 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Fabric - Applied Researcher, Cryptography Hardware

Fabric

Santa Clara, California, United States (Remote)
4 Months ago
Riot Games - Principal Software Engineer - Teamfight Tactics, Major Projects

Riot Games

Los Angeles, California, United States (On-Site)
3 Months ago
Microsoft - Research Intern - Reliable and Safe AI Agents

Microsoft

Redmond, Washington, United States (On-Site)
1 Month ago
Google - Student Researcher, BS/MS, Winter/Summer 2025

Google

(On-Site)
3 Months ago
NVIDIA - Signal and Power Integrity Engineer (RDSS Intern)

NVIDIA

Taipei City, Taiwan (On-Site)
1 Month ago
Luxoft - Senior Computer Systems Linux Engineer w/ Python

Luxoft

Bucharest, Bucharest, Romania (On-Site)
3 Months ago
NVIDIA - Deep Learning Performance Architect

NVIDIA

Beijing, Beijing, China (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