Europe Arbeitnow ·
full-timeAI Research Engineer (kernel & Inference Optimization)
Jobgether · UK
This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for an AI Research Engineer (Kernel & Inference Optimization) based in United Kingdom. You will work at the intersection of AI research, systems engineering, and high-performance model inference. Your focus will be on developing and optimizing model-serving architectures for advanced AI systems across a range of hardware environments. You will tackle challenges involving latency, throughput, memory efficiency, and scalability, including deployment on resource-constrained mobile and edge devices. The role combines hands-on research with low-level engineering, giving you the opportunity to develop novel inference strategies and GPU kernels. You will work with complex architectures spanning text, image, audio, diffusion models, and vision transformers. Your work will involve rigorous benchmarking, production testing, and iterative optimization to translate research into measurable performance improvements. You will collaborate with cross-functional teams in a highly technical, remote environment focused on pushing the boundaries of efficient AI systems. Accountabilities Design and deploy advanced model-serving architectures optimized for high throughput, low latency, and efficient memory utilization. Develop inference pipelines capable of operating effectively across diverse environments, including resource-constrained mobile devices and edge platforms. Establish clear performance targets covering response latency, token generation speed, throughput, memory footprint, and reliability. Build and execute controlled inference benchmarks in simulated and production environments, tracking latency, throughput, memory consumption, and error rates. Create and maintain representative datasets and simulation scenarios for evaluating model performance under real-world and resource-constrained conditions. Identify computational and memory bottlenecks across inference pipelines and implement solutions involving batching, networking, memory management, and other system-level optimizations. Develop custom GPU kernels and compute shaders for mobile hardware, including solutions written in Metal Shading Language (MSL). Apply advanced inference optimization techniques such as pruning, quantization, Flash Attention, KV caching, and speculative decoding. Design and optimize distributed inference systems using approaches such as tensor parallelism, pipeline parallelism, and expert parallelism for large-scale GPU workloads. Work with cross-functional engineering and research teams to integrate optimized inference frameworks into production and edge-device applications. Define evaluation methodologies, document experimental results, compare performance against established benchmarks, and continuously refine optimization strategies. Monitor production performance and use empirical research to identify opportunities for further improvements in scalability, efficiency, and reliability. Requirements: Degree in Computer Science or a related technical field; a PhD in NLP, Machine Learning, or a related discipline is highly relevant, particularly with a strong AI research track record and publications at leading conferences. Proven expertise in Metal Shading Language (MSL), including the ability to write custom compute shaders from scratch. Demonstrated experience with low-level kernel optimization and inference optimization on mobile or other resource-constrained devices. Track record of delivering measurable improvements in inference latency, throughput, and memory footprint for domain-specific applications. Deep understanding of modern model-serving architectures, inference engines, and optimization techniques for high-performance AI deployment. Strong experience writing GPU kernels for mobile devices such as smartphones. Practical experience developing and deploying end-to-end inference pipelines, from model optimization through production integration on constrained hardware. Strong ability to apply empirical research and systematic experimentation to solve latency, computational, and memory challenges. Experience designing robust evaluation and benchmarking frameworks for inference systems. Knowledge of distributed inference techniques, including tensor parallelism, pipeline parallelism, and expert parallelism for large-scale GPU clusters. Deep understanding of the mathematical foundations and architecture of diffusion models and Vision Transformers. Familiarity with modern inference optimization techniques including pruning, quantization, Flash Attention, KV Cache optimization, and speculative decoding such as EAGLE. Strong analytical and problem-solving abilities, with an ability to investigate complex system bottlenecks and turn research findings into practical engineering solutions. Excellent English communication skills and the ability to collaborate effectively with distributed, cross-functional technic
| Role | AI Research Engineer (kernel & Inference Optimization) |
|---|---|
| Company | Jobgether |
| Location | UK |
| Compensation | Not disclosed |
| Deadline | Rolling |
Typical process for this type of role
A general guide — the exact steps for this specific listing may vary; check the original posting for details.
- 1ApplicationSubmit your resume through the apply link.
- 2ScreeningRecruiter reviews your background against the role.
- 3AssessmentA technical test, assignment, or coding round, depending on the role.
- 4Interview(s)One or more rounds with the hiring team.
- 5OfferOffer letter with compensation and start date.
Before you apply
0/4More at Jobgether
Other jobs at Jobgether
- AI Video Editor · UK
- AI Architect / AI Engineering Lead · UK
- AI Trainers Network - Maltese · Switzerland
- AI Research Engineer (kernel & Inference Optimization) · France
- Account Manager (email Marketing) · France
Explore Related Placements
// similar opportunities
You might also like
AI Research Engineer (kernel & Inference Optimization)
Jobgether
This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for an AI Research Engineer (Kernel ...
Senior ML Engineer | Kimchi (LLM Inference Optimization)
Kimchi
Why Cast AI? Cast AI is an automation platform that operates cloud-native and AI infrastructure at scale. By embedding autonomous decision-making directly into ...
AI Researcher — Distillation
Jobgether
This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for a Linux Software Engineer - Ubun...
Senior Machine Learning Engineer, LLM Inference Optimization
Nebius
About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports develo...
AI Harness Engineer
Jobgether
This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for an AI Harness Engineer based in ...
AI Architect / AI Engineering Lead
Jobgether
This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for a AI Architect / AI Engineering ...