Skip to content

controny/matchstick

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages