| A | B | C | D | |
|---|---|---|---|---|
| 1 | Main | |||
| 2 | torch.amp | mixed precision | ||
| 3 | torch.autograd | |||
| 4 | torch.compile | Very important | https://pytorch.org/docs/stable/generated/torch.compile.html | |
| 5 | torch.cuda | |||
| 6 | torch.distributed | |||
| 7 | torch.func | |||
| 8 | torch.futures | |||
| 9 | torch.fx | |||
| 10 | torch.nn | |||
| 11 | torch.onnx | |||
| 12 | torch.random | |||
| 13 | torch.utils.benchmark | |||
| 14 | https://docs.google.com/document/d/1GgvOe7C8_NVOMLOCwDaYV1mXXyHMXY7ExoewHqooxrs/edit?pli=1#heading=h.fh8zzonyw8ng |