This repo implements a mini-version of the Torch API (that is why I call it Matchstick, as opposed to a torch), based on the schema by MiniTorch.
This toy library demonstrates many internal concepts underlying deep learning systems. Powered by Pyreverse, the key UML diagram of Tensor Functions is shown as follows.
To get started, first read setup to build your workspace.
Implemented features:
- GitHub CI
- Basic Neural Networks and Modules
- Autodifferentiation for Scalars
- Tensors, Views, and Strides
- Parallel Tensor Operations
- GPU / CUDA Programming in NUMBA
- Convolutions and Pooling
- Advanced NN Functions