TransformerEngine
..
README.md
Basic MNIST Example
python test_single_gpu_mnist.pypython test_single_gpu_mnist.py --use-te # Linear layers from TransformerEnginepython test_single_gpu_mnist.py --use-te --use-fp8 # FP8 + TransformerEngine for Linear layers