Software Engineering Intern, DLFW Comms - 2027

Responsibilities

• Integrate new communication libraries features in AI frameworks: from PoC to performance analysis to production.
• Perform deep analysis of AI workloads and frameworks to identify multi-GPU communication requirements and opportunities.
• Collaborate hands-on with teams working on the latest AI models.
• Author custom communication or fused compute-communication kernels to showcase ultimate performance on NV platforms.
• Conduct in-depth research to achieve SOL GPU performance.
• Build fault-tolerant and elastic solutions for large-scale or dynamic AI workloads.
• Collaborate with a very dynamic team across multiple time zones.
• What we need to see
• You are pursuing a M.S. or Ph.D. in CE/CS/EE with a strong background in communication, kernel authoring, and/or AI training/inference.
• Rapid prototyping and development with Python, C++, CUDA or related DSLs (Triton, cuTe).

Languages