MixedPrecisionPolicy

class tensorplay.distributed.fsdp.MixedPrecisionPolicy(param_dtype: Any = None, reduce_dtype: Any = None, output_dtype: Any = None, cast_forward_inputs: bool = True)[source]