Network performance engineer optimizing NIC, PCIe, driver, kernel, and RDMA communication for single- and multi-host AI inference clusters — improving KV cache transfer, cross-node collective communication, and model-weight loading, while troubleshooting packet loss and latency. Core stack: Linux, RDMA (RoCEv2/InfiniBand), C/C++, plus Python/Go/Shell.
You will optimize NIC, PCIe, driver, kernel, and RDMA communication performance across single-host and multi-host AI inference clusters. You will improve critical inference communication paths, including KV cache transfer, cross-node collective communication, and model-weight loading. You will investigate packet loss, latency variation, slow nodes, and NIC failures; strengthen monitoring and automated diagnostics; collaborate with inference-framework, hardware, and NIC-vendor partners; and evaluate emerging networking technologies.
Responsibilities
Optimize NIC, PCIe, driver, kernel, and RDMA communication performance