Glossary · Phase 4 · Deep learning
PyTorch
The most widely used Python library for building and training neural networks.
Try it now
Read shapes from the outside in: batch, position, features. Most tensor bugs are a dimension in the wrong slot.
Learn it properly
PyTorch tensors · NumPy-like arrays that run on GPUs. About 30 minutes.
Other terms from the same lesson
- CUDA: NVIDIA's system for running code on its GPUs. In PyTorch, "cuda" is the name of the GPU device.
Never trained an AI?
Teach a small one to write, right in your browser. It takes about 20 seconds and needs no account.