Project description
Luxoft is looking for an AI software development engineer to develop ML kernels in the Triton kernel language. We are looking for an engineer who is passionate about optimizing Machine Learning GPU kernels and improving the performance of key applications and benchmarks. What you do directly impacts the performance of AMD GPUs and enables us to become a competitive solution for generative AI. Become a part of our high-impact and incredibly talented Triton kernels team.
Responsibilities
Develop ML kernels for matrix multiplication, Flash Attention and other ML operators
Benchmark, perform competitive analysis and optimize kernels to improve performance
Collaborate with the GPU architecture team to improve future generations
Apply knowledge of software engineering best practices
Skills
Must have
Proficiency with C/C++
Proficiency in CUDA or HIP / ROCm or OpenCL programming
Solid understanding of parallel programming models, and optimization techniques
Strong problem-solving skills and the ability to work in a collaborative environment
Nice to have
Familiarity with models like LLama, Mixtral and Gemma is a plus
Knowledge of MLIR, LLVM and GPU assembly and GPU architecture is a plus
Familiarity with PyTorch or JAX
Other
Languages
English: B2 Upper Intermediate
Luxoft Benefits*
- ●Relocation options **
- ●Experience in an international environment
- ●Cross-cultural experience
- ●Feedback culture
- ●Regular appraisals
- ●Annual holiday - 20 or 26 days. The duration of the leave depends on the overall seniority
- ●Occasional leave - 1 or 2 days/ depending on the circumstances
- ●Child care leave - 2 days or 16 hours per year
- ●Absence due to force majeure - 2 days or 16 hours per year
- ●Maternity Leave - 20 weeks
- ●Parental Leave - 41 weeks
- ●Paternity Leave - 14 days
- ●Expert-led tech courses covering basic to advanced topics
- ●Internal instructor-led soft skills courses
- ●Comprehensive in-house self-learning resources for both soft and hard skills
- ●Access to external self-learning libraries like ProQuest eBook and Udemy for Business
- ●Cloud Programs: MS Cloud Academy, AWS Partner Academy, Google Cloud Academy
- ●Custom Learning Programs: upskilling, reskilling, technical mentorship
- ●Leadership Programs for Managers
- ●Multisport card
- ●Possibility to order Multisport card at the corporate rate for family members
- ●LuxGood Program: wellbeing seminars, contests, relaxation sessions, yoga sessions, etc.
- ●One Team Program: Buddy for each New Joiner; seminars, meeting and workplace space to support integration with local community and culture; “Hire me” workshops for partners
- ●Preferential banking offer
- ●Preferential car leasing offer
- ●Cafeteria program discounts for shops, cinema tickets, holiday offers
- ●Luxoft Social Benefit Fund: sport and recreation benefits, the possibility to receive financial support
- ●Private Healthcare Insurance with unlimited access to specialists
- ●Full dental support
- ●Travel Insurance
- ●Possibility to add private healthcare coverage for family members at the corporate rate
- ●Life insurance at the corporate rate for employees and family members, including payment of the basic package for the employee by the employer
- ●Reimbursement for corrective glasses
- ●Many fun social activities organized by the Luxoft team offline in your city
- ●Online entertainment events for whole company and local team events
- ●A workplace where you’re treated with respect within a multicultural team
- ●Rotation between projects and accounts
- ●New career opportunities
*The acquisition of rights to the above benefits depends on the form of cooperation. Benefits apply to those employed under a contract of employment.
**Please note that relocation is not available for all open positions. At Luxoft Poland it is possible to work remotely only from the territory of Poland.
***Options offered by the Polish government.
Apply for Senior Kernel Developer in Remote Poland
*Indicates a required field