PyTorch (torch, torch.tensor)

The spelled-out intro to language modeling: building makemore

Andrej Karpathy

Deep learning framework used to store counts in tensors, perform matrix operations, sample with torch.multinomial, and run training/backprop.