def METHOD_NAME(self):
    """Tests pure FP16 training, including when the parameter's dtype is
    changed after FSDP initialization and before training."""
    self.run_subtests(
        {
            "cpu_offload": [
                CPUOffload(offload_params=True),
                CPUOffload(offload_params=False),
            ]
        },
        self._test_pure_fp16_training,
    )