Software Engineer - AI Networking & Distributed GPU Systems

Meta; Menlo Park (CA); On-site; USD 154,000 - 217,000; Full time

Benefits offered by this job

Bonus

Equity

Benefits

Job summary

Meta is seeking a Software Engineer for the AI Networking Software team, focused on NCCL-based distributed GPU communication for large-scale ML workloads. You will help design, optimize, and own software across multi-GPU and multi-node training stacks, enabling GenAI/LLM scaling with high performance and reliability.

You will lead technical initiatives, collaborate with ML engineers and infrastructure teams, and contribute to performance benchmarks and optimizations across CUDA and PyTorch

Qualifications

• Bachelor's degree in Computer Science, Computer Engineering, or equivalent practical experience.
• Proven C/C++ and Python programming skills.
• Proven track record of leading successful projects.

Responsibilities

• Provide technical leadership for the collective communication library development on Meta's large-scale GPU training infra with a focus on GenAI/LLM scaling.
• Lead cross-functional technical projects and communicate decisions to technical and non-technical stakeholders.
• Collaborate to optimize performance and reliability of the AI networking software stack across NCCL and PyTorch integration.

Skills

C/C++

Python

Leadership

Distributed systems

Education

Bachelor's degree in CS/CE

PhD in CS/CE (preferred)

Tools

NCCL

CUDA

PyTorch

Job description

Meta is seeking a Software Engineer for the AI Networking Software team, focused on NCCL-based distributed GPU communication for large-scale ML workloads. You will help design, optimize, and own software across multi-GPU and multi-node training stacks, enabling GenAI/LLM scaling with high performance and reliability.

You will lead technical initiatives, collaborate with ML engineers and infrastructure teams, and contribute to performance benchmarks and optimizations across CUDA and PyTorch