Ray Direct Transport (RDT) API

Usage with Core APIs

Enable RDT for actor tasks with the @ray.method decorator, or pass _tensor_transport to ray.put(). You can then pass the resulting ray.ObjectRef to other actor tasks, or use ray.get() to retrieve the result. See Ray Direct Transport (RDT) for more details on usage.

ray.method	Annotate an actor method.
ray.put	Store an object in the object store.
ray.get	Get a remote object or a list of remote objects from the object store.

Collective tensor transports

Collective tensor transports require a collective group to be created before RDT objects can be used. Use these methods to create and manage collective groups for the gloo and nccl tensor transports.

ray.experimental.collective.create_collective_group	Create a collective group on the given list of actors.
ray.experimental.collective.get_collective_groups	Get the collective groups that the given actors are a subset of.
ray.experimental.collective.destroy_collective_group	Destroy a collective group.