Principal Engineer - AI Networking

About the Role

The ideal candidate is an experienced RDMA software engineer with a strong background in high-performance networking, distributed communication systems, and systems programming. You will work closely with senior technical leaders to design, implement, optimize, and operate critical networking infrastructure used by large-scale AI training and inference workloads. This is a hands-on engineering role requiring deep technical expertise, strong software development skills, and a passion for solving complex performance and scalability challenges.

Responsibilities

• Design, develop, and optimize RDMA-based software components and services for large-scale AI infrastructure.
• Build and enhance collective communication frameworks, transport layers, and communication libraries used by distributed AI workloads.
• Develop congestion management, load balancing, resiliency, and failover capabilities for RDMA-based networks.
• Analyze and improve communication performance across networking, GPU, and software stacks.
• Design and implement scalable distributed systems supporting AI training and inference environments.
• Collaborate with networking, AI infrastructure, hardware, and cloud platform teams to deliver high-performance solutions.
• Investigate and resolve complex networking, performance, and reliability issues in production environments.
• Develop observability, telemetry, debugging, and performance analysis tools for distributed communication systems.
• Contribute to architectural design discussions and technical direction for networking platforms.
• Participate in code reviews and help maintain engineering excellence across the team.

Requirements

• Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, or related field; advanced degree preferred.
• 7+ years of software engineering experience in systems software, networking, distributed systems, or infrastructure platforms.
• Strong hands-on expertise with RDMA technologies, including RoCEv2 and/or InfiniBand.
• Experience developing RDMA-enabled software, communication libraries, networking services, or distributed infrastructure.
• Strong understanding of RDMA programming concepts, including queue pairs, completion queues, memory registration, verbs, and transport semantics.
• Proficiency in C/C++ and Linux systems programming.
• Experience debugging and optimizing performance-critical software systems.
• Solid understanding of networking fundamentals, operating systems, and distributed systems concepts.

Preferred Qualifications

• Experience with collective communication frameworks and libraries such as NCCL, RCCL, MPI, UCX, UCC, XCCL, or similar technologies.
• Experience supporting AI/ML infrastructure and distributed training environments.
• Knowledge of GPUDirect RDMA and GPU-aware communication technologies.
• Experience developing congestion management, traffic engineering, or network resiliency solutions.
• Familiarity with large-scale GPU clusters and high-performance computing environments.
• Experience building services and infrastructure operating directly over RDMA transports.
• Familiarity with distributed training frameworks such as PyTorch, DeepSpeed, Megatron-LM, TensorFlow, or JAX.
• Experience with Kubernetes, containers, and cloud infrastructure platforms.
• Understanding of performance profiling and benchmarking tools for networking and distributed systems.

Pay

US: Hiring Range in USD from: $114,600 - $234,600 per year. May be eligible for bonus, equity, and compensation deferral.

Benefits

• Medical, dental, and vision insurance, including expert medical opinion
• Short term disability and long term disability
• Life insurance and AD&D
• Supplemental life insurance (Employee/Spouse/Child)
• Health care and dependent care Flexible Spending Accounts
• Pre-tax commuter and parking benefits
• 401(k) Savings and Investment Plan with company match
• Paid time off: Flexible Vacation is provided to all eligible employees assigned to a salaried (non-overtime eligible) position. Accrued Vacation is provided to all other employees eligible for vacation benefits. For employees working at least 35 hours per week, the vacation accrual rate is 13 days annually for the first three years of employment and 18 days annually for subsequent years of employment. Vacation accrual is prorated for employees working between 20 and 34 hours per week. Employees working fewer than 20 hours per week are not eligible for vacation.
• 11 paid holidays
• Paid sick leave: 72 hours of paid sick leave upon date of hire. Refreshes each calendar year. Unused balance will carry over each year up to a maximum cap of 112 hours.
• Paid parental leave
• Adoption assistance
• Employee Stock Purchase Plan
• Financial planning and group legal
• Voluntary benefits including auto, homeowner and pet insurance