Could you give an example? My thought was something similar to using torch.distributed’s TCPStore, something like
store = torch.distributed.TCPStore(..)
rand_tensor = torch.rand((64, 10))
store.set('rand_tensor1', rand_tensor)

but this errors because set() only takes str as the second arg.