README.md 286 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
# Basic MNIST Example with optional FP8

```bash
python mnist.py
python mnist.py --use-te   # Linear layers from TransformerEngine
python mnist.py --use-fp8  # FP8 + TransformerEngine for Linear layers
```

# Benchmark A Basic Transformer Layer

```bash
python transformer_layer.py
```