Description
We are seeking a Senior Software Architect to join our AI Systems and Networking team. As a Senior Software Architect, you will be responsible for owning modules and projects end-to-end, from scoping research questions to shipping production code. This role requires a recognized expert who drives technical decisions, pulls in ideas from research and industry, and regularly prototypes new approaches to prove a point.
Key responsibilities include:
- Architecting and implementing high-performance communication and memory management libraries for distributed AI
- Driving hardware-software co-optimization with GPU, DPU, NIC, and switch teams through GPUDirect RDMA, NVLink, and next-generation interconnects
- Profiling and optimizing data movement across GPU memory, system DRAM, NVMe, and network fabrics
- Integrating networking capabilities into AI serving stacks such as vLLM, SGLang, and TensorRT-LLM
To be successful in this role, you will need to have 8+ years of experience in systems software and/or networking with demonstrated ownership of complex projects. You should also have a solid understanding of high-performance networking, including InfiniBand, RoCE, RDMA, NVLink, and GPUDirect. Additionally, you should be proficient in C/C++/Rust systems programming and have a strong understanding of ML systems concepts.
If you are a motivated and experienced software architect looking to join a dynamic team, please apply for this exciting opportunity.