pub fn write_stage_message(
writer: impl Write,
message: &StageWireMessage,
dtype: WireActivationDType,
) -> Result<()>pub fn write_stage_message(
writer: impl Write,
message: &StageWireMessage,
dtype: WireActivationDType,
) -> Result<()>