/Simple-DeepLearning-System

A Simple Deep Learning System

Primary LanguagePythonMIT LicenseMIT

Simple-DeepLearning-System

mit license

Introduction

This is a simple deep learning system with numpy. It include a autograd system which is the basis of the whole system while all network layers and tensor operations that implemented are based on the simple dynamic computation graph. We now support fundamental mathmatics operation and several layers, including Linear, ReLU, Dropout1d, BatchNorm1d, Softmax, and some pre-defined loss functions such as MSELoss and CrossEntropyLoss. We plan to support some core components of convolution nerual networks and CUDA in the future.

Quick Start