Remote Senior AI Software Engineer

Job Description

The Senior AI Software Engineer will design and develop AI communication middleware for distributed workloads, focusing on performance-critical features and optimizing distributed training. Responsibilities also include profiling workloads, enhancing GPU communication paths, and collaborating with cross-disciplinary teams.

Requirements: Candidates should have over 8 years of experience in high-performance systems programming, particularly in C/C++ on Linux, with strong expertise in GPU communication stacks. Experience with AI frameworks and open-source contributions is preferred.

Key Skills: C/C++, Linux, GPU Communication Stacks, CUDA, ROCm, NCCL, RCCL, Distributed Training, Collective Communication, Open-source Contributions, AI Frameworks, libfabric, OFI, UCX, RDMA, RoCEv2

Benefits: Health Benefits, Flexible Work Options