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 performance improvements for neural networks using Triton on ROCm-enabled GPUs. Strong C/C++ and GPGPU experience (HIP, CUDA, OpenCL) are essential, along with compiler internals knowledge (LLVM, GCC). Performance analysis skills are highly valued.
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:
  • Basic understanding of ML technologies
  • Experience with PyTorch
  • LLVM and MLIR 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

G5 Games - 2D UI/UX Artist (match-3 project)

G5 Games

Astana, Astana, Kazakhstan (Remote)
4 Months ago
Homa games - Senior Machine Learning Engineer

Homa games

Paris, Île-de-France, France (On-Site)
7 Months ago
Match Group - Sr. Software Engineer, Generative AI

Match Group

Palo Alto, California, United States (Hybrid)
4 Months ago
Ello - Tech Lead, Machine Learning

Ello

San Francisco, California, United States (On-Site)
3 Months ago
Intel Corporation - SW Embedded Engineer

Intel Corporation

Haifa, Haifa District, Israel (On-Site)
3 Months ago
NVIDIA - Senior Architect, GPU and SoC Modeling

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Google - Senior Software Engineering Manager, Google Cloud

Google

Hyderabad, Telangana, India (On-Site)
1 Month ago
Cadence - Principal Software Engineer

Cadence

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

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Ello - Tech Lead, Machine Learning

Ello

San Francisco, California, United States (On-Site)
3 Months ago
Intel Corporation - AI Frameworks Engineer

Intel Corporation

Ireland (Hybrid)
2 Months ago
Luxoft - Senior/Lead Machine Learning and Image Processing Specialist

Luxoft

Poland, Ohio, United States (Remote)
3 Months ago
Paypal - Sr. Staff Machine Learning Scientist

Paypal

San Jose, California, United States (Hybrid)
4 Months ago
Playrix - Game Designer

Playrix

Armenia (Remote)
4 Months ago
NVIDIA - Manager, Tools and Development

NVIDIA

Pune, Maharashtra, India (On-Site)
1 Month ago
Paypal - Machine Learning Manager

Paypal

San Jose, California, United States (Hybrid)
3 Months ago
Tencent - Research Intern (NLP)

Tencent

Palo Alto, California, United States (On-Site)
2 Days ago
Tencent - Senior Staff Researcher

Tencent

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

Playrix

Montenegro (Remote)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Mexico City, Mexico City, Mexico

Keywords Studios (Player Support) - Videogames Customer Service (Written Support)

Keywords Studios (Player Support)

Benito Juárez, Estado De México, Mexico (Hybrid)
1 Month ago
Rackspace Technology - Presales Data Science Architect – AWS Cloud

Rackspace Technology

Mexico City, Mexico (On-Site)
3 Months ago
Netflix - Team Lead, Department Assistants

Netflix

Mexico City, Mexico City, Mexico (On-Site)
1 Month ago
Crunchyroll - Staff Site Reliability Engineer

Crunchyroll

Mexico City, Mexico City, Mexico (On-Site)
3 Months ago
Paypal - Account Executive

Paypal

Mexico City, Mexico City, Mexico (On-Site)
4 Months ago
Amber - Game Engineer

Amber

Guadalajara, Jalisco, Mexico (On-Site)
7 Months ago
The Walt Disney Company - Account Executive, DASP

The Walt Disney Company

Mexico City, Mexico (On-Site)
3 Weeks ago
Rackspace Technology - Policy Management Specialist  (Security Risk & Compliance)

Rackspace Technology

Puebla, Puebla, Mexico (On-Site)
3 Months ago
Eleven Labs - Sales Development Representative - Mexico

Eleven Labs

Mexico (Remote)
1 Month ago
Codeninja - Azure Engineer

Codeninja

Mexico (Remote)
1 Month ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Riot Games - Associate Art Director - League of Legends, Game Modes

Riot Games

Sydney, New South Wales, Australia (On-Site)
8 Months ago
Microsoft - Senior Silicon Architect

Microsoft

Noida, Uttar Pradesh, India (On-Site)
1 Month ago
Ceragon Networks - Senior Verification Engineer

Ceragon Networks

Karnataka, India (On-Site)
3 Months ago
NVIDIA - Senior VLSI Physical Design Integration Engineer

NVIDIA

Massachusetts, United States (On-Site)
1 Month ago
Tesla - Bachelor/Master Thesis Research and Development, Mechanical Engineering

Tesla

Prüm, Rhineland-Palatinate, Germany (On-Site)
1 Week ago
Tesla - Mechanical Design Engineer Internship

Tesla

Brandenburg, Germany (On-Site)
1 Week ago
NVIDIA - Senior ASIC Power Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Virtuos - Senior Technical Director

Virtuos

France (Hybrid)
4 Months ago
NVIDIA - Senior Circuit Design Engineer

NVIDIA

Austin, Texas, United States (Hybrid)
2 Days ago
Fabric - Applied Cryptographer, ZKP Research

Fabric

Los Angeles, California, United States (Remote)
4 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