Remote Position

[Remote] Principal Product Manager, Augmented Memory Grid (AMG)

🌍 Remote · United States 💼 Full-time

Note: The job is a remote job and is open to candidates in USA. WEKA is building NeuralMesh, an AI-native data infrastructure platform for enterprise reasoning workloads. The Principal Product Manager will own the Augmented Memory Grid roadmap and go-to-market, partnering with engineering, enterprise customers, GPU cloud providers, and inference-stack partners to improve large-scale LLM inference performance, efficiency, and cost.

Responsibilities

  • Own the AMG product roadmap: KV-cache/prefix-cache offload, memory tiering, and integration with inference engines and orchestration layers (vLLM, NVIDIA Triton/TensorRT-LLM/NIM, Kubernetes-based serving)
  • Partner with engineering to define architecture trade-offs across GPU memory, networking (RDMA, GPUDirect, NVMe-oF), and distributed storage — translating inference performance bottlenecks (time-to-first-token, throughput, context length) into product requirements
  • Work directly with enterprise customers and GPU cloud partners: Nebius, CoreWeave, TogetherAI, etc., running production inference workloads to gather requirements, validate benchmarks, and prioritize features that reduce cost-per-token and improve SLAs at scale
  • Partner with NVIDIA and other silicon/inference-stack partners on joint roadmap and certification work
  • Define and track benchmarks (TTFT, throughput, cache hit rate) that demonstrate AMG's value versus standard GPU-memory-only inference
  • Support sales and field teams with technical positioning, competitive differentiation, and enterprise deal support

Skills

  • Strong leadership skills with a history of successfully leading cross-functional teams
  • You are a strategic thinker who can develop and execute product strategies that align with market trends and customer needs, as well as think critically about existing strategies
  • You have a proven ability to translate strategic goals into actionable plans and deliver results
  • You have excellent communication and interpersonal skills, with the ability to articulate complex technical concepts to both technical and non-technical stakeholders
  • You are comfortable presenting product strategies and roadmaps to internal teams and external customers
  • Hands-on product or engineering experience with LLM inference serving — vLLM, NVIDIA Triton/TensorRT-LLM/NIM, Ray Serve, or comparable — and fluency in concepts like KV-cache, prefix/context caching, quantization, and batching strategies
  • Working knowledge of how modern LLMs are served in production (context windows, multi-tenant serving, GPU scheduling) well enough to translate model-level constraints into infrastructure requirements
  • Comfort with the fundamentals of high-performance networking and distributed systems — RDMA, GPUDirect Storage, NVMe-oF, or equivalent — and how they affect inference performance
  • Track record working directly with large enterprise accounts — requirements gathering, production deployments, SLAs — not solely self-serve/PLG products
  • 10+ years of product management experience, ideally with some portion in infrastructure, ML platforms, or developer-facing technical products
  • Prior experience at a GPU cloud, inference platform, or AI infrastructure startup
  • Familiarity with storage systems (parallel/distributed file systems, object storage) in AI/ML pipelines
  • Experience partnering directly with NVIDIA or other accelerator/silicon vendors

Company Overview

  • WEKA is a global data platform company that delivers a cloud-native, software-based data platform for AI & next-generation workloads It was founded in 2013, and is headquartered in Campbell, California, USA, with a workforce of 201-500 employees. Its website is http://www.weka.io.
  • Company H1B Sponsorship

  • WEKA has a track record of offering H1B sponsorships, with 2 in 2026, 2 in 2025. Please note that this does not guarantee sponsorship for this specific role.
  • Apply Now →

    More Open Positions

    Explore similar remote opportunities