Senior Applied Scientist, Efficient LLM Inference & Model Optimization

About the role

Nebius Token Factory is seeking an Applied Scientist to join our team. This role involves designing and executing research projects that lead to productionized inference capabilities, collaborating with engineers, and publishing credible work.

Responsibilities

• Own focused research projects from hypothesis through experiment, ablation, prototype, and production handoff.
• Prepare internal reports, technical blogs, or papers when the work is externally credible.
• Partner directly with Machine Learning Engineers (MLEs) to ensure research prototypes become usable production components.
• Define and execute research programs in efficient Large Language Models (LLMs) and Vision Language Models (VLMs) inference with measurable production impact.
• Invent, evaluate, and productionize methods for quantization, QAT, distillation, speculative decoding, KV-cache reuse, KV-cache compression, long-context inference, MoE routing, and model/runtime co-optimization.
• Build high-quality prototypes in PyTorch, Triton, CUDA-adjacent tooling, or inference-serving frameworks, then work with MLEs and platform engineers to productionize them.
• Design rigorous evaluation methodology covering quality, latency, throughput, numerical stability, memory footprint, tail latency, and cost per token.
• Publish papers, technical reports, blog posts, and open-source artifacts that build external credibility for Nebius Token Factory.
• Collaborate with MLE, GPU kernel, backend infrastructure, product, and customer teams to choose high-leverage research bets.
• Mentor engineers and scientists on experimental design, scientific rigor, and model/system tradeoffs.

Requirements

• PhD in computer science, machine learning, ML systems, computer systems, computer architecture, electrical engineering, applied math, or a closely related field.
• Strong publication record or equivalent research artifacts in ML, ML systems, efficient inference, model compression, quantization, distillation, serving systems, or related areas.
• Strong hands-on coding ability in Python and PyTorch; ability to move from idea to experiment to prototype quickly.
• Deep understanding of LLMs, VLMs, transformer inference, decoding algorithms, model compression, quantization, and production-serving tradeoffs.
• Strong experimental design skills, including ablations, baselines, metrics, statistical reasoning, and failure analysis.
• Excellent written and verbal communication.

Qualifications

• First-author publications in NeurIPS, ICML, ICLR, MLSys, ACL, EMNLP, ASPLOS, OSDI, SOSP, ISCA, HPCA, or comparable venues.
• 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 with post-training, SFT, DPO, RLHF, RLAIF, preference optimization, or synthetic data generation when connected to inference quality or efficiency.
• Open-source research artifacts, widely used benchmarks, high-quality technical blogs, or invited talks in efficient AI systems.

Skills

• Strong experimental design skills.
• Ability to work independently and collaboratively.
• Knowledge of industry standards and best practices in AI and ML.
• Experience with relevant tools and technologies.

Benefits

• 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.

Pay

Base Compensation Range: $195,200—$262,200 USD

Schedule

Full-time