Software Engineer, ML Networking

About the role

A systems-level engineer specializing in network infrastructure and network optimization, with expertise in building and maintaining software that interfaces with networks. You will be responsible for writing and maintaining software that interfaces between our accelerators and our high-speed networks. This role requires deep technical knowledge of network protocols, kernel-space and/or user-space networks, interfacing with hardware, and the ability to debug and optimize distributed software at the network level.

Networking Systems Engineering

• Expert-level proficiency with network protocols and networking concepts
• Deep kernel networking: TCP/IP stack internals, XDP, eBPF, io_uring, and epoll
• User-space networking: DPDK, RDMA, kernel bypass techniques
• Understanding of how to build higher-level abstractions like collectives and RPC
• Skilled at diagnosing and resolving networking issues in distributed systems, especially at OSI model layers 2-4
• Low-Level Systems And OS Programming
• Strong programming skills in a systems programming language, including memory management, lock-free data structures, and NUMA-aware programming
• Software, driver, and OS performance optimization tools and techniques
• Comfort with or desire to learn Rust
• Strong Candidates May Have
• Understanding of ML accelerators and accelerator drivers
• Demonstrated ability to design new network protocols
• Experience with PCIe and drivers for PCIe devices
• Expertise in algorithms used in networking, including compression and graph algorithms
• Experience programming on SmartNICs
• 5+ years of experience in systems programming or network programming

Representative Projects

• Build a system for accelerator-initiated tensor movement over the network
• Benchmark software for a new networking environment
• Implement a new collective algorithm to improve latency
• Optimize congestion control algorithms for large-scale synchronous workloads
• Debug kernel-level network latency spikes

Pay

The annual compensation range for this role is $280,000—$850,000 USD.

Logistics

• Minimum education: Bachelor’s degree or an equivalent combination of education, training, and/or experience
• Required field of study: A field relevant to the role as demonstrated through coursework, training, or professional experience
• Minimum years of experience: Years of experience required will correlate with the internal job level requirements for the position
• Location-based hybrid policy: Currently, we expect all staff to be in one of our offices at least 25% of the time. However, some roles may require more time in our offices.
• Visa sponsorship: We do sponsor visas! However, we aren't able to successfully sponsor visas for every role and every candidate. But if we make you an offer, we will make every reasonable effort to get you a visa, and we retain an immigration lawyer to help with this.