Jobs in Stevenage, United Kingdom
Senior Kernel Developer
Luxoft
Research Scientist, Machine Learning – Search & Recommendations
NLP PEOPLE
Talent Acquisition Specialist
Tradeify
Senior DevOps Engineer
Public Sector Resourcing, managed by AMS
Sr Commodities Business Analyst
Luxoft
CEO (Supply Chain Software)
MBR Partners
Senior Compensation Analyst
Instacart
Connect with headhunters to apply for similar jobsMortgage Case Manager (Part-time) - Stirling
JohnstonGreer Ltd
Lead Auditor for Quality, Environment and Occupational Health & Safety systems
DNV Germany Holding GmbH
Marketing Implementer
MSP Marketing Edge
Head of Client Services
Media Meerkat
SuccessFactors product Owner CGEMJP00315973
Experis - ManpowerGroup
Construction Manager Large Projects
Hitachi Vantara Corporation
Data entry Administrators (Ongoing Project)
Michael Page (UK)
Mortgage Case Manager (Part-time) - Leeds
JohnstonGreer Ltd
Mortgage Administrator (Part-time) - York
JohnstonGreer Ltd
Sales Executive (Part Time /Remote)
Colville Search Limited
Senior Product Analyst - Enterprise Product Operations
Wise
Employment Legal Counsel
Taylor Root
Remote Head of APAC - Infrared Finance
Blockchain Works
Remote Engineering Manager - NALA
WorksHub
UX Delivery Manager/Architect
Luxoft
Senior Kernel Developer
Job summary
A technology consulting company in the United Kingdom is seeking an AI software development engineer to develop and optimize ML kernels. This role involves working on matrix multiplication and other operators, collaborating with GPU architecture teams, and applying best practices in software engineering. The ideal candidate has strong proficiency in C/C++ and CUDA programming, as well as solid problem-solving skills. The position offers a dynamic and impactful work environment.
Qualifications
- Must have proficiency with C/C++.
- Proficient in CUDA or HIP/ROCm or OpenCL programming.
- Solid understanding of parallel programming models and optimization techniques.
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.
Skills
Tools
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.
- 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
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
* The salary benchmark is based on the target salaries of market leaders in their relevant sectors. It is intended to serve as a guide to help Premium Members assess open positions and to help in salary negotiations. The salary benchmark is not provided directly by the company, which could be significantly higher or lower.
job faster
Follow us
