Senior GPU Compiler Software Development Engineer

About the Position

We are seeking a Senior GPU Compiler Software Development Engineer to work on GPU support for OpenAI/Triton, a language and compiler for efficient custom Deep-Learning primitives. You will be collaborating 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 the OpenAI/Triton project for GPUs.
• Communication with other developers, customers, and project managers.
• Test implementation, project documentation, and verification of the system with unit/component/functional tests.

Requirements

• Must have

  • Strong C/C++ programming skills
  • 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 GPGPU (General purpose GPU) computing (HIP, CUDA, OpenCL, etc.)
  • 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

We Offer

• Remote work from Mexico
• Competitive salary
• Opportunity to work in the Automotive Industry

About the Company

Luxoft is a global provider of software and digital solutions, empowering businesses to innovate and thrive in the digital age. We are committed to delivering excellence and fostering a collaborative and inclusive work environment.

Apply

If you are passionate about GPU development and want to make a significant impact, we encourage you to apply for this exciting opportunity.