Search Remote Jobs

Senior Machine Learning Engineer, LLM Inference Optimization

Job not on LinkedIn

đŸ”„ 17 hours ago

đŸ‡ȘđŸ‡ș Europe – Remote

⏰ Full Time

🟠 Senior

đŸ—Łïž LLM Engineer

đŸ‘» Ghost score 25%

infoinfo
Apply Now
Find Similar Remote Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Logo of Nebius Group

Nebius Group

1001 - 5000 employees

đŸ€– Artificial Intelligence

🏱 Enterprise

☁ SaaS

Artificial Intelligence ‱ Enterprise ‱ SaaS

Nebius Group is building one of the world’s leading AI infrastructure companies, focusing on providing the necessary compute, storage, and tools for developers in the AI space. Based in Europe and listed on Nasdaq, Nebius has a global presence with R&D centers across Europe, North America, and Israel. The company's primary offering is an AI-centric cloud platform designed for intensive AI workloads, complemented by various other businesses involved in generative AI development, edtech, and autonomous technology.

📋 Description

‱ Own optimization work for specific model families, customer endpoints, or serving backends ‱ Run engine comparisons and recommend practical serving configurations for specific workloads ‱ Debug model quality or performance regressions during production rollouts ‱ Optimize LLM and VLM endpoints for latency, throughput, memory efficiency, GPU utilization, quality, and cost per token ‱ Deploy, configure, benchmark, and extend inference engines such as vLLM, SGLang, TensorRT-LLM, Triton Inference Server, NVIDIA Dynamo, or similar systems ‱ Build and productionize model-compression workflows, including quantization, quantization-aware training, distillation, low-bit serving, and accuracy recovery ‱ Implement or integrate speculative decoding, draft-model approaches, KV-cache optimization, prefix caching, chunked prefill, continuous batching, and disaggregated prefill/decode serving ‱ Build reproducible benchmark harnesses for TTFT, TPOT, tokens per second per GPU, p95/p99 latency, GPU memory, reliability, and cost per token ‱ Partner with GPU kernel engineers and platform engineers to diagnose bottlenecks across model code, kernels, runtime, scheduler, gateway, and cluster layers ‱ Write clear design docs, performance reports, rollout plans, and customer-facing technical explanations

🎯 Requirements

‱ Strong Python and PyTorch engineering skills ‱ Hands-on experience deploying or optimizing LLM, VLM, or high-throughput transformer inference systems ‱ Practical knowledge of at least one modern inference stack such as vLLM, SGLang, TensorRT-LLM, Triton Inference Server, NVIDIA Dynamo, Ray Serve, KServe, or equivalent internal systems ‱ Strong understanding of transformer inference bottlenecks, including KV cache, attention, memory bandwidth, batching, parallelism, and long-context serving ‱ Ability to reason quantitatively about latency, throughput, quality, utilization, and cost tradeoffs ‱ Strong communication skills and ability to collaborate with research, kernel, infrastructure, product, and customer teams ‱ Experience with quantization-aware training, post-training quantization, FP8, INT8, INT4, NVFP4, MXFP4, AWQ, GPTQ, SmoothQuant, or related techniques ‱ Experience with distillation, speculative decoding, EAGLE, Medusa, multi-token prediction, or other inference acceleration methods ‱ Experience with agentic workloads, including tool calling, structured outputs, streaming APIs, high concurrency, and multi-step orchestration ‱ CUDA or Triton familiarity ‱ Open-source contributions to vLLM, SGLang, TensorRT-LLM, FlashInfer, LMCache, PyTorch, Triton, Ray, KServe, or related projects

đŸ–ïž Benefits

‱ Competitive compensation ‱ Career growth and learning opportunities ‱ Flexibility and ownership ‱ Collaborative and innovative culture ‱ Opportunity to work on impactful AI projects ‱ International environment and talented teams

Apply Now

Similar Jobs

🕒 July 8

CrackenAGI

11 - 50

🔒 Cybersecurity

đŸ€– Artificial Intelligence

☁ SaaS

Fullstack Engineer with a focus on Generative AI, developing a proactive cyber platform. Collaborate with top-tier experts in a fully remote setup.

đŸ‡ȘđŸ‡ș Europe – Remote

⏰ Full Time

🟡 Mid-level

🟠 Senior

đŸ—Łïž LLM Engineer