🟠Join us in Luxoft!
🏦Flexible working hours
🩺Private Medical & Dental care & Life Insurance
💰 Paid Referrals
🏋🏽 ♀️ MyBenefit program (sports card, well-being program etc.)
🌎 Internal Mobility program - possibility of rotation between projects, locations, accounts
🎓 LuxTalent platform (webinars, training, courses)
🟠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
What is relevant to have
• Strong C/C++ programming skills
• Experience with compiler internals (llvm, gcc or any other)
• Basic Python programming skills
• Experience in performance analysis
What is 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
🟠Languages
English: B2 Upper Intermediate
🟠Seniority
Senior
🟠Work mode
hybrid in Warsaw, Wroclaw, Gdansk or Krakow