TE EP comm backend with NCCL EP #3021

Open

Assignees

opened

on May 21, 2026

Add an EP communication backend to TransformerEngine via NCCL EP, exposed through a single nvte_* C API to both PyTorch and JAX, and working on any RDMA NIC.

Metadata

Assignees

phu0ngng

Labels

No labels

No labels

Type

Fields

No fields configured for Epic.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests