The library described in this section is available on GitHub.

The project serves as an experimental work for:

  • designing a non-trivial machine learning framework.
  • implementing low-level primitives, on top of mature libraries such as cuDNN and cuBLAS.
  • exposing Python bindings for common workflows.

Its long-term goal is to implement classical CNN architectures such as LeNet, AlexNet, GoogLeNet, VGGNet, and ResNet without relying on high-level deep learning frameworks.