M

Marvell

High-performance semiconductor solutions for data infrastructure

Systems Performance Principal Engineer - AI Solutions Lab

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSanta Clara, CA, USAIn-person

About the Job

Requirements

  • Experience with AI serving frameworks such as vLLM, SGLang, TensorRT-LLM, or similar platforms.
  • Strong background in RCCL/NCCL, RDMA, communication-compute overlap, and performance profiling.
  • Hands-on experience with GPU profiling tools such as Nsight Systems/Compute, PyTorch Profiler, or ROCm Compute Profiler to localize kernel-level bottlenecks.
  • Ability to diagnose end-to-end bottlenecks across application, framework, runtime, and infrastructure layers.
  • Working knowledge of at least one distributed training framework, such as PyTorch DDP/FSDP, DeepSpeed, or Megatron-LM.
  • Strong Python and Linux skills.
  • Comfort working in containerized environments such as Docker to ensure benchmark reproducibility.
  • Ability to translate technical analysis into clear recommendations and concise performance narratives.
  • Bachelor's or Master's degree in Computer Science, Computer Engineering, or a related field.

Responsibilities

  • Drive end-to-end performance optimization across AI serving and training stacks using leading models and customer-relevant configurations.
  • Identify bottlenecks across the full data path, from application request flow through compute, memory, communication layers, GPU kernels, network transport, and data/storage pipelines.
  • Develop and execute benchmark methodologies that are reproducible, technically credible, and relevant to customer use cases.
  • Profile serving-stack internals, including quantization, scheduling, KV-cache management, and disaggregated prefill/decode, to identify concrete optimization targets.
  • Analyze results and develop performance narratives that guide kernel, runtime, communication, and system-level optimizations.
  • Work with NRE/SRE and network or fabric architects to close competitive gaps through targeted tuning.
  • Create performance summaries, visuals, and benchmark artifacts for customer presentations, executive reviews, and trade-show demonstrations.

Desired Qualifications

  • Go experience.
  • Kubernetes experience.
  • Experience benchmarking distributed large-model inference and training systems.
  • Familiarity with GPU kernel analysis, system tracing, and workload-specific optimization.
  • Familiarity with serving-stack internals such as continuous batching, paged KV-cache attention, and quantization techniques such as GPTQ and SmoothQuant.
  • Exposure to customer proofs of concept, competitive benchmarking, or externally consumed performance reporting.
  • A PhD.

About the Company

Marvell

High-performance semiconductor solutions for data infrastructure

Marvell Technology, Inc. creates high-performance semiconductor products that power data infrastructure for telecommunications operators, data centers, and enterprises. Its offerings span computing, storage, and networking to enable efficient, secure data transmission, storage, and processing. The products are programmable and scalable platforms designed for high bandwidth and strong security, supporting 5G networks and the broader digital economy. Revenue comes from designing, manufacturing, licensing, and providing related services to other businesses that integrate these components into their own products. Unlike many peers, Marvell emphasizes programmable, scalable platforms tailored to data infrastructure needs and long-term partnerships with enterprise and telecom customers. The company aims to help customers upgrade their networks and data systems to increase capacity, performance, and efficiency while expanding its own business in the data infrastructure space.

Company size
10,001+
Company stage
IPO
Headquarters
Santa Clara, CA
Founded
1995

Recently Posted Jobs

Explore more opportunities at Marvell

See more jobs