Senior GPU Compiler Software Development Engineer

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

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

Playrix - Game Designer

Playrix

Ireland (Remote)
3 Months ago
DISCO - Engineering Manager , Feature

DISCO

Gurugram, Haryana, India (On-Site)
3 Months ago
ByteDance - Machine Learning Research Scientist, AI for Science

ByteDance

Seattle, Washington, United States (On-Site)
1 Month ago
Paypal - Sr. Staff Machine Learning Scientist

Paypal

San Jose, California, United States (On-Site)
4 Months ago
Canva - Senior Computer Vision Engineer - Photo AI

Canva

Vienna, Vienna, Austria (Remote)
3 Days ago
GlobalLogic - Embedded C IRC238457

GlobalLogic

Chennai, Tamil Nadu, India (Hybrid)
4 Months ago
Fabric - Applied Researcher, Cryptography Proof Systems

Fabric

France (Remote)
3 Months ago
Riot Games - Staff Software Engineer (Build) - Teamfight Tactics

Riot Games

Los Angeles, California, United States (On-Site)
1 Week ago
The Walt Disney Company - Lead Machine Learning Engineer

The Walt Disney Company

Washington, United States (On-Site)
1 Day ago
Neostella - Senior Graph Database Architect

Neostella

Medellín, Antioquia, Colombia (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Axinous - Sr. Staff Machine Learning Engineer

Axinous

San Jose, California, United States (Hybrid)
1 Month ago
CynLr - Software Engineer - GPU performance

CynLr

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Netflix - Machine Learning Engineer

Netflix

United States (Remote)
4 Weeks ago
PwC - Conversational AI Developer- Manager

PwC

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Paypal - Senior Manager, Machine Learning

Paypal

San Jose, California, United States (Hybrid)
4 Months ago
DeepSight AI Labs   - Intern/Computer Vision Engineer

DeepSight AI Labs

Gurugram, Haryana, India (On-Site)
8 Months ago
Playrix - Game Director

Playrix

Ireland (Remote)
3 Months ago
Samsung Semiconductor - Intern, Machine Learning Engineer - VLMs

Samsung Semiconductor

San Jose, California, United States (Hybrid)
2 Weeks ago
Playrix - Feature Owner (LiveOps)

Playrix

Georgia (Remote)
3 Months ago
ByteDance - Engineering Manager - Applied Machine Learning Algorithm

ByteDance

San Jose, California, United States (On-Site)
3 Months 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

Cadence - Senior DFT Engineer

Cadence

Bengaluru, Karnataka, India (On-Site)
4 Months ago
RoofStack - Senior Software Developer (Payment Systems)

RoofStack

İstanbul, İstanbul, Türkiye (Hybrid)
1 Month ago
The Walt Disney Company - Principal Ride Development Engineer

The Walt Disney Company

Lake Buena Vista, Florida, United States (On-Site)
2 Days ago
Rivos - CPU Design Verification - Full-time

Rivos

Bengaluru, Karnataka, India (Hybrid)
3 Months ago
Assystems - Mechanical Engineer

Assystems

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Samsung Semiconductor - Intern, CXL Deep Memory Research Scientist

Samsung Semiconductor

San Jose, California, United States (Hybrid)
1 Week ago
The Walt Disney Company - Lead Machine Learning Engineer

The Walt Disney Company

Washington, United States (On-Site)
1 Day ago
Riot Games - Executive Recruiting Partner

Riot Games

Los Angeles, California, United States (On-Site)
4 Weeks ago
ByteDance - Lead Research Scientist, Foundation Model, Speech & Audio

ByteDance

San Jose, California, United States (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)

Chennai, Tamil Nadu, India (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