Applied scientist in efficient LLM inference

Nebius; Netherlands; Vor Ort; EUR 175.000 - 235.000; Vollzeit

Benefits dieser Stelle

Health insurance

401(k) plan

Parental leave

Remote work reimbursement

Disability & life insurance

Competitive compensation

Career growth

Ownership and impact

Zusammenfassung

Nebius Group is leading a new era in cloud infrastructure for the global AI economy. We build an end-to-end AI cloud platform for data and model workflows, from training to production, designed for engineers.

The selected scientist will drive research on efficient LLM/VLM inference, develop prototypes with PyTorch, Triton, CUDA tooling, and collaborate with engineers to turn findings into scalable production components. Expect a fast-paced, collaborative environment.

Qualifikationen

• PhD in computer science, ML systems, or closely related discipline.
• Strong Python and PyTorch implementation skills, turning ideas into experiments and prototypes.
• Deep knowledge of LLMs, VLMs, transformer inference, decoding algorithms, and production-serving trade-offs.
• Strong experimental design skills with ablations, baselines, metrics, and failure analysis.
• First-author publications at top venues are a plus.

Aufgaben

• Lead research projects in efficient LLM and VLM inference from hypotheses to production handoff.
• Develop and evaluate methods like quantization, QAT, distillation, and MoE routing.
• Build prototypes using PyTorch, Triton, CUDA-adjacent tooling and collaborate to production components.
• Research LLM request routing and scheduling strategies including cache-aware load balancing.
• Evaluate latency, throughput and cost across queueing, KV-cache transfer, and worker placement.
• Study multi-node inference and WideEP for dense and mixture-of-experts models.
• Share results through internal reports, blogs, papers, and mentor others on experimental design.

Kenntnisse

Python

PyTorch

LLMs knowledge

VLMs knowledge

Transformer inference

Experiment design

Communication skills

Publications

Ausbildung

PhD in computer science or related field

Tools

TensorRT-LLM

Triton

CUDA

PyTorch internals

SGLang

FlashAttention

Jobbeschreibung

Описание

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 developers and enterprises from data and model training through to production deployment, without the cost and complexity of building large in-house AI/ML infrastructure. Built by engineers, for engineers. From large-scale GPU orchestration to inference optimization, we own the hard problems across compute, storage, networking and applied AI. Nebius Token Factory is looking for scientists who can turn frontier LLM and VLM inference bottlenecks into research questions, conduct rigorous experiments, and translate their findings into production capabilities.

Задачи

• Lead research projects in efficient LLM and VLM inference, from hypotheses and experiments through ablations, prototypes, and production handoff.
• Develop and evaluate methods spanning quantization, quantization‑aware training (QAT), distillation, speculative decoding, KV‑cache reuse, KV‑cache compression, long‑context inference, MoE routing, and model/runtime co‑optimization.
• Build prototypes using PyTorch, Triton, CUDA‑adjacent tooling, or inference‑serving frameworks, and collaborate with MLEs and platform engineers to turn them into production components.
• Research LLM request routing and scheduling strategies, including cache‑aware load balancing and prefill–decode disaggregation (PDD).
• Evaluate how queueing, KV‑cache transfer, worker placement, and prefill/decode capacity allocation affect latency, throughput, and serving cost.
• Investigate multi‑node inference for dense and mixture‑of‑experts models, including wide expert parallelism (WideEP).
• Study expert placement, load imbalance, and computation/communication trade‑offs, and use the findings to guide model and system architecture choices.
• Develop rigorous evaluations covering quality, latency, throughput, numerical stability, memory footprint, tail latency, and cost per token.
• Compare serving architectures under consistent workload conditions and GPU budgets.
• Work with MLE, GPU kernel, backend infrastructure, product, and customer teams to select research priorities with measurable production impact.
• Share results through internal reports, technical blogs, papers, and open‑source artifacts, and mentor engineers and scientists on experimental design, scientific rigor, and model/system trade‑offs.

Требования

• A PhD in computer science, machine learning, ML systems, computer systems, computer architecture, electrical engineering, applied mathematics, or a closely related discipline.
• A strong publication record or equivalent research artifacts in ML, ML systems, efficient inference, model compression, quantization, distillation, serving systems, or related areas.
• Strong Python and PyTorch implementation skills, with the ability to turn ideas into experiments and working prototypes.
• Deep knowledge of LLMs, VLMs, transformer inference, decoding algorithms, model compression, quantization, and production‑serving trade‑offs.
• Strong experimental design skills covering ablations, baselines, metrics, statistical reasoning, and failure analysis.
• Excellent written and verbal communication.
• Будет плюсом: First‑author publications at venues such as NeurIPS, ICML, ICLR, MLSys, ACL, EMNLP, ASPLOS, OSDI, SOSP, ISCA, or HPCA;
• Experience deploying ML models or inference optimizations in production;
• Experience with vLLM, SGLang, TensorRT‑LLM, NVIDIA Dynamo, FlashAttention, FlashInfer, Triton, CUDA, or PyTorch internals;
• Experience applying post‑training, SFT, DPO, RLHF, RLAIF, preference optimization, or synthetic data generation to inference quality or efficiency;
• Open‑source research artifacts, widely used benchmarks, technical blogs, or invited talks demonstrating contributions to efficient AI systems;
• Research or implementation experience in one or more areas of distributed inference system architecture, such as LLM request routing, PDD, multi‑node serving, or MoE expert parallelism, including WideEP.

Условия

• Health insurance: 100% company‑paid medical, dental, and vision coverage for employees and families.
• 401(k) plan: Up to 4% company match with immediate vesting.
• Parental leave: 20 weeks paid for primary caregivers, 12 weeks for secondary caregivers.
• Remote work reimbursement: Up to $85/month for mobile and internet.
• Disability & life insurance: Company‑paid short‑term, long‑term and life insurance coverage.
• We offer competitive compensation and benefits packages.
• Base Compensation Range $195,200 — $262,200 USD.
• 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.
• Fast moving - Bold thinking - Constant growth - Meaningful impact - Trust and real ownership - Opportunity to shape the future of AI.