Base Career helps you apply smarter for this job.
Key skills for this role
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of models and hardware, a position that took years to build.
Inferact's mission is to grow vLLM as the world's AI inference engine and accelerate AI progress by making inference cheaper and faster. Founded by the creators and core maintainers of vLLM, we sit at the intersection of models and hardware, a position that took years to build.
We're looking for exceptional University of Waterloo co-op students who want to work on the systems that determine how fast, efficiently, and reliably frontier AI models run on frontier workloads at scale. This is not a sandboxed internship project. We will match your strengths and interests to a real engineering problem across the vLLM stack, from model execution and low-level accelerator code to distributed serving and the cloud platform that makes it all usable.
You'll work alongside the creators and core maintainers of vLLM on work intended to ship into open source, production systems, or the tooling that supports both. You will have a primary technical track, meaningful ownership, and mentorship from a small, senior team, with opportunities to collaborate across models, compilers, accelerators, networking, and distributed systems. The goal is to let exceptional students learn at the frontier while making contributions used by developers and AI teams around the world.
Your co-op will have a primary home in one of the following tracks, with opportunities to contribute across others:
Inference Runtime: Bring new model architectures and inference techniques to life in vLLM. Implement ideas from research papers; support mixture-of-experts, multimodal, diffusion, and agentic workloads; and improve scheduling, continuous batching, KV-cache memory management, prefix caching, and hybrid model serving.
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
San Francisco, USA
Performance & Scale: Build the distributed serving data plane that lets vLLM run across many GPUs and nodes. Work on tensor, expert, or context parallelism; prefill/decode separation and KV-cache transport; fault tolerance and multi-tenancy; and high-performance communication using NCCL, DeepEP, NVSHMEM, RDMA, or InfiniBand.
Kernel Engineering: Raise the hardware performance ceiling by writing and optimizing attention, GEMM, sampling, KV-cache, fused, and quantization kernels. Use CUDA, Triton, TileLang, CUTLASS/CUTE, or related tools; reason about memory hierarchy, occupancy, and tensor cores; and prove speedups through profiling, correctness tests, and reproducible benchmarks.
AMD GPU Performance: Help make vLLM first-class on AMD accelerators. Work across ROCm, HIP, Triton, CK, AITER, kernels, runtime paths, quantization, compiler integration, and performance-regression infrastructure while learning how AMD-specific execution, memory, and toolchain constraints shape inference.
TPU Performance: Help make vLLM fast and correct on Google TPUs. Build backend, runtime, and compiler integrations with JAX, XLA, Pallas, MLIR, and related tooling; inspect compiler artifacts; work on lowering, fusion, and code generation; and benchmark production-relevant serving across correctness, latency, and throughput.
Cloud Orchestration: Build the operational platform that makes large-scale inference deployable and reliable. Work on Kubernetes and custom operators, topology-aware GPU scheduling, zero-downtime vLLM rollouts, token-aware routing, observability, infrastructure-as-code, automated recovery, and bring-your-own-cloud or multi-cloud fleet management.
You are not expected to arrive with experience in every track. We care most about unusual depth or learning velocity in one area, strong fundamentals, and evidence that you can turn a difficult problem into working, measurable software.
Location: San Francisco, California. This co-op is in-office only at Inferact's San Francisco office and is intended for the University of Waterloo co-op program.
Work term: Co-op / internship. Exact dates will align with the applicable Waterloo work term.
Compensation: Competitive compensation based on the applicable co-op market and candidate background, plus a housing stipend for the duration of the co-op term.
Inferact is a private AI infrastructure company building open-source software that makes LLM inference faster and cheaper.
Visit company websiteJobs and hiring trendsInternship
Entry
Onsite
Apply faster on company sites with our extension.