Using FP8 mixed-precision in accelerate give error. Where and how do I install the necessary packages?

As per title, I'm experimenting with 8gb gpu training and would like to use fp8 precision for speed improvement but I have this error:

ValueError: Using `fp8` precision requires `transformer_engine` or `MS-AMP` to be installed.

Is there a way to install this packages?

Start by reviewing the FP8 precision handling and the installation guidance for `transformer_engine` or `MS-AMP`. Document the supported package installation steps and any relevant requirements so users can resolve the reported ValueError; done means the FP8 setup instructions are clear and reproducible.

Written by the indexing model from the issue text.