Build and optimize the local AI inference stack for RTX, RTX Pro, and DGX GPUs, with a focus on performance, stability, and scalability across diverse hardware architectures.
Architect and develop modern inference runtimes and execution stacks using frameworks such as llama.cpp, vLLM, PyTorch, WinML, DXCGC, and TensorRT-RTX, supporting LLM, vision-language, TTS, ASR, and diffusion-based AI workloads.
Perform end-to-end optimization of AI models, data pipelines, and inference runtimes to maximize performance on current and next-generation GPU architectures. Apply model optimization techniques, including quantization, pruning, sparsity, and distillation, to enable efficient deployment of large models on local and edge devices.
Lead system-level debugging, performance tuning, and performance-accuracy trade-off analysis; develop infrastructure for performance and accuracy sweeps; analyse results to identify gaps and drive fixes; and establish engineering guidelines to accelerate bring-up and ensure production readiness of new models and inference backends.
Mentor other engineers and review architecture proposals spanning several groups, acting as a force multiplier for technical quality and consistency across the broader LocalAI organization.
What we need to see
5+ Years of experience with Bachelor’s, Master’s, or PhD in Computer Science, Software Engineering, Mathematics, or related field, or equivalent experience.
Excellent C++ programming and debugging skills, with a strong foundation in data structures, algorithms, and machine learning.
Proven experience architecting and optimizing AI inference pipelines and applications using ML/DL frameworks such as Llama.cpp, vLLM, PyTorch, Windows ML, DXCGC, and TensorRT.
Deep understanding of inference backends and runtime internals, including scheduling, memory management, KV-cache behavior, graph execution, quantization, and hardware-aware optimization.
Demonstrated ability to establish technical direction and influence outcomes across multiple teams and organizations without relying on formal management authority.
Strong analytical and problem-solving skills, with the ability to manage multiple priorities effectively in a fast-paced environment.
Excellent written and verbal communication skills, enabling effective collaboration across engineering teams and management.
Ways to stand out from the crowd
Deep understanding of modern machine learning, deep neural network, and generative AI techniques, with exposure to AOT graph-compilation systems (TensorRT builder, MLIR-based compilers)
Consistent track record of delivering end-to-end products in multinational companies with geographically distributed teams.
Expert-level low-level systems and GPU kernel programming, CUDA, and development of high-performance systems.
Significant contributions to open-source inference runtimes, model tooling, or performance infrastructure, as a maintainer or core contributor.
Hands-on experience building applications using frameworks and APIs such as llama.cpp, PyTorch, TensorRT, Vulkan, DirectX, and vLLM.
We're a top employer recognized for innovation, growth, and commitment to diversity as an equal opportunity workplace. We offer competitive salaries, a generous benefits package, and the opportunity to work alongside some of the industry's most talented and forward-thinking professionals. As our engineering teams continue to grow rapidly, we're looking for creative, self-driven engineers with a passion for technology to join us.
