ColossalAI/examples/tutorial/fp8/mnist/README.md

Basic MNIST Example with optional FP8

python main.py
python main.py --use-te   # Linear layers from TransformerEngine
python main.py --use-fp8  # FP8 + TransformerEngine for Linear layers