AI Infrastructure Engineer

Posted 2 Days Ago
2 Locations
In-Office
Junior
Artificial Intelligence • Hardware • Semiconductor • Generative AI
The Role
Own and optimize Netpreme’s production LLM inference infrastructure, including vLLM, SGLang, TensorRT-LLM, NVIDIA Dynamo, Kubernetes deployments, benchmarking, distributed serving, KV caching, batching, routing, and scheduling. The role focuses on improving latency, throughput, scalability, reliability, and efficiency across diverse LLM and multimodal workloads while partnering with hardware, model, and systems engineers.
Summary Generated by Built In
About Netpreme

The biggest bottleneck in AI inference today is memory. Chips can compute faster than they can be fed data, a problem known as the memory wall, and Netpreme is building the solution. Our approach spans the full stack, from a proprietary memory management library down to X-Mem™, our Memory Processing Unit (MPU) silicon. Netpreme was co-founded by researchers in optical networking and ML systems. We’re a small team in Cambridge, MA and Santa Clara, CA, and we’re looking for people who want to own hard problems end to end and help shape where the company goes next.

About the Role

Serving frontier LLMs efficiently at scale is a moving target: every new model and workload reshuffles the tradeoffs across parallelism, batching, KV caching, and scheduling. You’ll own Netpreme’s inference-serving stack end to end, from tuning vLLM, SGLang, and TensorRT-LLM to architecting distributed serving with NVIDIA Dynamo (prefill/decode disaggregation, KV-cache-aware routing). Working directly with our hardware and systems engineers, you’ll build the benchmarks and drive the optimizations that take new models from research to production.

This role will be performed onsite from one of our offices in Santa Clara, CA or Boston, MA (relocation assistance available).

The Opportunity
  • Impact: AI is advancing faster than the infrastructure underneath it. We’re working at that layer, unlocking more capability from every GPU while dramatically improving efficiency. The work you do here compounds across state-of-the-art models, systems, and real-world applications.

  • Timing: Joining now means real ownership, including equity, and a direct say in product direction. You’ll work from first principles, move quickly from insight to execution, and see your work show up in what we ship, not just in a roadmap.

  • Culture: We care about rigor, clarity, and impact. Disagreement is welcome when it’s thoughtful, the best idea wins regardless of who it comes from, and you’ll learn something new most weeks.

Your Impact
  • Deploy and operate models in production: Run large language and multimodal models using inference engines like vLLM, SGLang, and TensorRT-LLM.

  • Own the serving infrastructure: Build and maintain distributed inference systems with NVIDIA Dynamo, Kubernetes, Helm, and Ansible, including upgrades, monitoring, troubleshooting, and reliability.

  • Optimize performance: Tune parallelism strategies (TP/EP/DP/PP), batching, KV caching, routing, and scheduling to meet latency, throughput, and SLA targets.

  • Benchmark and measure: Build reproducible benchmarks for latency (TTFT, TPOT, E2E), throughput, concurrency scaling, and system efficiency.

  • Support diverse workloads: Enable tool calling, agentic and coding workflows, RAG, and long-context applications.

  • Bring new tech into production: Evaluate new models, frameworks, and optimizations, and partner with hardware, model, and systems engineers to ship improvements.

Qualifications
  • BS, MS, or PhD in Computer Science, Computer Engineering, or a related field, or equivalent practical experience.

  • 2+ years of engineering experience in LLM inference, ML systems, GPU systems, or related infrastructure.

  • Hands-on experience deploying and optimizing at least one LLM inference engine (e.g., vLLM, SGLang, or TensorRT-LLM).

  • Solid understanding of LLM inference fundamentals: prefill/decode, batching, KV caching, and latency/throughput tradeoffs.

  • Strong Python and Linux skills, plus hands-on experience with Kubernetes and containerized deployments.

  • Comfortable benchmarking and troubleshooting systems, and communicating clearly in a small engineering team.

Nice to Have
  • Experience with Helm, Ansible, or similar deployment and configuration tools.

  • Experience with distributed inference systems like NVIDIA Dynamo, including parallelism strategies (TP/EP/DP/PP), prefill/decode disaggregation, and KV-cache-aware routing.

  • Experience with advanced inference optimizations such as prefix caching, speculative decoding, and quantization.

  • Experience analyzing LLM architectures and tuning deployments across different hardware platforms to meet SLAs.

  • Contributions to vLLM, SGLang, TensorRT-LLM, or related open-source projects.

  • Experience owning production infrastructure, ideally at an early-stage startup.

What We Offer
  • Competitive salary with performance-based bonus and early-stage equity grant

  • 100% employer-paid Health, Dental, and Vision coverage for you and your dependents

  • 401(k) match with immediate vesting, and access to financial advisors to help you reach your financial goals

  • 100% employer-paid Life, Disability, and AD&D insurance, plus a fitness stipend and wellness & mental health perks

  • Generous PTO: 20 vacation days, 15 company holidays (including 3 floating days of your choosing)

  • Daily lunch stipend

  • Enterprise-level Claude & ChatGPT access with a generous token budget

  • Well-equipped, sunny offices in Santa Clara, CA & Cambridge, MA with on-site parking and EV charging; on-site fitness center in Santa Clara; gym discounts near our Cambridge office

  • Relocation assistance available to work in Santa Clara, CA or Boston, MA

  • Visa sponsorship and relocation assistance to one of our office hubs

  • A collaborative, continuous-learning environment with smart, dedicated colleagues building the next generation of high-performance computing architecture

Skills Required

  • BS, MS, or PhD in Computer Science, Computer Engineering, or a related field, or equivalent practical experience
  • 2+ years of engineering experience in LLM inference, ML systems, GPU systems, or related infrastructure
  • Hands-on experience deploying and optimizing at least one LLM inference engine, such as vLLM, SGLang, or TensorRT-LLM
  • Solid understanding of LLM inference fundamentals, including prefill/decode, batching, KV caching, and latency/throughput tradeoffs
  • Strong Python and Linux skills
  • Hands-on experience with Kubernetes and containerized deployments
  • Ability to benchmark and troubleshoot systems and communicate clearly in a small engineering team
  • Experience with Helm, Ansible, or similar deployment and configuration tools
  • Experience with distributed inference systems such as NVIDIA Dynamo, including TP, EP, DP, PP, prefill/decode disaggregation, and KV-cache-aware routing
  • Experience with prefix caching, speculative decoding, and quantization
  • Experience analyzing LLM architectures and tuning deployments across different hardware platforms to meet SLAs
  • Contributions to vLLM, SGLang, TensorRT-LLM, or related open-source projects
  • Experience owning production infrastructure, ideally at an early-stage startup
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
10 Employees
Year Founded: 2024

What We Do

Netpreme develops memory-to-compute interconnect solutions for generative AI systems, aiming to expand memory bandwidth and capacity, enhance efficiency, and reduce costs.

Similar Jobs

Remote or Hybrid
US
15100 Employees
78K-108K Annually

Capital One Logo Capital One

Artificial Intelligence Engineer

Fintech • Machine Learning • Payments • Software • Financial Services
Hybrid
5 Locations
55000 Employees
230K-286K Annually

NVIDIA Logo NVIDIA

Support Engineer

Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
In-Office
4 Locations
21960 Employees
108K-207K Annually

NVIDIA Logo NVIDIA

Infrastructure Engineer

Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
In-Office or Remote
5 Locations
21960 Employees
184K-357K Annually

Similar Companies Hiring

Unusual Machines, Inc. Thumbnail
Hardware • Robotics
Orlando, FL
300 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
33 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
65 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account