Senior AI/LLM Inference Engineer

Persistent; Bengaluru; Ibrido; INR 4.000.000 - 8.000.000; Tempo pieno

Vantaggi offerti da questo lavoro

Competitive salary

Culture focused on talent development

Flexible work hours

Long Service awards

Annual health check-ups

Insurance coverage: group term life,个人

Descrizione del lavoro

Persistent is seeking a Senior AI/LLM Inference Engineer in Bengaluru to design, optimize, and deploy high-performance inference solutions for large language and multimodal models across CPU, GPU, and edge environments.

You will lead evaluation, benchmarking, RAG experimentation, and quantization strategies while collaborating with researchers and platform engineers to productionize cutting-edge LLM capabilities. Hybrid work options are available.

Competenze

• 5 to 10 years of experience in AI Systems Engineering or similar fields.
• Strong programming in Python and modern C++ with debugging, profiling, memory management, and multithreaded development.
• Deep understanding of Transformer architectures, attention mechanisms, tokenization, embeddings, positional encoding, KV cache optimization, and text generation techniques.
• Hands-on experience with PyTorch, Hugging Face Transformers, and modern LLM development ecosystems.
• Practical experience with LLMs such as Llama, Qwen, Mistral, Gemma, GPT-family models, or equivalent foundation models.
• Strong expertise in LLM evaluation methodologies, benchmarking frameworks, and model performance measurement.
• Experience implementing Retrieval-Augmented Generation (RAG), prompt engineering, and model fine-tuning solutions.

Mansioni

• Design and optimize high-performance inference pipelines for LLMs across CPU, GPU, and edge platforms.
• Build and maintain evaluation frameworks for reasoning, factuality, instruction following, and multimodal capabilities.
• Benchmark foundation models using reproducible datasets and quantitative metrics.
• Experiment with prompt engineering, RAG, fine-tuning, and inference-time scaling.
• Develop automated evaluation pipelines and bias assessment mechanisms.
• Optimize model latency, throughput, memory, and operational efficiency across deployment environments.
• Implement model quantization, mixed-precision inference, and memory optimization.
• Develop and optimize SIMD-based kernels using ARM NEON, AVX2, AVX-512, or SSE.
• Collaborate with researchers and engineers to productionize LLM capabilities.
• Analyze bottlenecks with profiling and performance engineering.

Conoscenze

Python

C++

Transformer architectures

Profiling

Strumenti

PyTorch

Hugging Face Transformers

CUDA

OpenCL

ONNX

Descrizione del lavoro

About Position:

We are seeking a highly skilled Senior AI/LLM Inference & GPU Optimization Engineer to design, optimize, evaluate, and deploy high-performance Large Language Model (LLM) and multimodal AI inference solutions across CPU, GPU, and edge environments. The ideal candidate will combine strong expertise in LLM architectures with systems programming, SIMD optimization, GPU acceleration, quantization techniques, and inference performance engineering. You will play a key role in building scalable AI systems that deliver low-latency, high-throughput inference while maintaining model accuracy and reliability.

• Role: Senior AI/LLM Inference Engineer
• Location: Bengaluru
• Experience: 5 to 10 Years
• Job Type: Full-Time Employment

What You’ll Do:

• Design and optimize high-performance inference pipelines for Large Language Models and multimodal foundation models across CPU, GPU, and edge platforms.
• Build and maintain evaluation frameworks for assessing reasoning, factuality, instruction following, long-context performance, robustness, tool usage, and multimodal capabilities.
• Compare and benchmark foundation models using reproducible datasets, quantitative metrics, qualitative analysis, and failure-mode categorization.
• Design experiments for prompt engineering, Retrieval-Augmented Generation (RAG), fine-tuning strategies, inference-time scaling, and model behavior analysis.
• Develop automated evaluation pipelines, including LLM-as-a-Judge methodologies, human validation workflows, and bias assessment mechanisms.
• Optimize model latency, throughput, memory utilization, and operational efficiency across various deployment environments.
• Implement model quantization strategies, mixed-precision inference, weight compression, and memory optimization techniques.
• Develop and optimize custom operators, inference kernels, and runtime components for AI workloads.
• Create vectorized CPU implementations leveraging ARM NEON, AVX2, AVX-512, and SSE instruction sets.
• Collaborate with AI researchers, platform engineers, and product teams to productionize cutting-edge LLM capabilities.
• Analyze inference bottlenecks using advanced profiling and performance engineering methodologies.
• Deploy and manage production-grade AI inference systems with reliability, scalability, and observability best practices.
• Stay current with advancements in AI acceleration, LLM architectures, inference frameworks, and hardware optimization technologies.

Expertise You’ll Bring:

• 5 to 10 years of experience in AI Systems Engineering, Machine Learning Infrastructure, High-Performance Computing, Systems Programming, or AI Platform Engineering.
• Strong programming expertise in Python and modern C++ with experience in debugging, profiling, memory management, and multithreaded development.
• Deep understanding of Transformer architectures, attention mechanisms, tokenization, embeddings, positional encoding, KV cache optimization, and text generation techniques.
• Hands-on experience with PyTorch, Hugging Face Transformers, and modern LLM development ecosystems.
• Practical experience working with LLMs such as Llama, Qwen, Mistral, Gemma, GPT-family models, or equivalent foundation models.
• Strong expertise in LLM evaluation methodologies, benchmarking frameworks, and model performance measurement.
• Experience implementing Retrieval-Augmented Generation (RAG), prompt engineering, and model fine-tuning solutions.
• Strong understanding of quantization techniques, mixed precision inference, calibration methodologies, weight packing, and numerical accuracy validation.
• Hands-on experience developing and optimizing SIMD-based kernels using ARM NEON, x86 AVX2, AVX-512, or SSE instruction sets.
• Experience with GPU acceleration technologies including CUDA, Metal, OpenCL, DirectML, or equivalent GPU compute frameworks.
• Strong understanding of GPU architectures, memory hierarchies, parallel computing concepts, and AI acceleration techniques.
• Experience with model deployment, inference runtimes, ONNX, model export workflows, and AI serving platforms.
• Proficiency with Linux development environments, Git, CMake, Ninja, compilers, linkers, shared libraries, and ABI compatibility.
• Experience building scalable CI/CD pipelines and production-grade AI platforms.
• Strong analytical, problem-solving, and performance optimization skills.

Benefits:

• Competitive salary and benefits package
• Culture focused on talent development with quarterly growth opportunities and company-sponsored higher education and certifications
• Opportunity to work with cutting-edge technologies
• Employee engagement initiatives such as project parties, flexible work hours, and Long Service awards
• Annual health check-ups
• Insurance coverage: group term life, personal accident, and Mediclaim hospitalization for self, spouse, two children, and parents

Values-Driven, People-Centric & Inclusive Work Environment:

Persistent is dedicated to fostering diversity and inclusion in the workplace. We invite applications from all qualified individuals, including those with disabilities, and regardless of gender or gender preference. We welcome diverse candidates from all backgrounds.

• We support hybrid work and flexible hours to fit diverse lifestyles.
• Our office is accessibility-friendly, with ergonomic setups and assistive technologies to support employees with physical disabilities.
• If you are a person with disabilities and have specific requirements, please inform us during the application process or at any time during your employment

"Persistent is an Equal Opportunity Employer and prohibits discrimination and harassment of any kind."