/pytorch-numpy

Pytorch implementation in NumPy

Primary LanguagePython

Pytorch in NumPy

Implementation of pytorch functionality in numpy

  • Autograd
  • Ready layers
  • Gradchecks
  • MLP models
  • Trainer

Layers implemented:

  • Linear
  • ReLu
  • Tanh
  • Dropout
  • Sequential
  • BatchNorm
  • Identity
  • Residual