README.md
setup.cfg
setup.py
edutorch/__init__.py
edutorch.egg-info/PKG-INFO
edutorch.egg-info/SOURCES.txt
edutorch.egg-info/dependency_links.txt
edutorch.egg-info/top_level.txt
edutorch/losses/__init__.py
edutorch/losses/softmax.py
edutorch/losses/svm.py
edutorch/losses/temporal_softmax.py
edutorch/nn/__init__.py
edutorch/nn/batchnorm.py
edutorch/nn/conv2d.py
edutorch/nn/dropout.py
edutorch/nn/embedding.py
edutorch/nn/layernorm.py
edutorch/nn/linear.py
edutorch/nn/lstm.py
edutorch/nn/lstm_cell.py
edutorch/nn/maxpool2d.py
edutorch/nn/module.py
edutorch/nn/relu.py
edutorch/nn/rnn.py
edutorch/nn/rnn_cell.py
edutorch/nn/spatial_batchnorm.py
edutorch/nn/spatial_groupnorm.py
edutorch/nn/temporal_linear.py
edutorch/nn/functional/__init__.py
edutorch/optim/__init__.py
edutorch/optim/adam.py
edutorch/optim/optimizer.py
edutorch/optim/rmsprop.py
edutorch/optim/sgd.py
edutorch/optim/sgd_momentum.py
tests/__init__.py
tests/gradient_check.py
tests/train_test.py
tests/losses/__init__.py
tests/losses/softmax_test.py
tests/losses/svm_test.py
tests/losses/temporal_softmax_test.py
tests/nn/__init__.py
tests/nn/batchnorm_test.py
tests/nn/conv2d_test.py
tests/nn/dropout_test.py
tests/nn/embedding_test.py
tests/nn/layernorm_test.py
tests/nn/linear_test.py
tests/nn/lstm_cell_test.py
tests/nn/lstm_test.py
tests/nn/maxpool2d_test.py
tests/nn/module_test.py
tests/nn/relu_test.py
tests/nn/rnn_cell_test.py
tests/nn/rnn_test.py
tests/nn/spatial_batchnorm_test.py
tests/nn/spatial_groupnorm_test.py
tests/nn/temporal_linear_test.py
tests/optim/__init__.py
tests/optim/adam_test.py
tests/optim/optimizer_test.py
tests/optim/rmsprop_test.py
tests/optim/sgd_momentum_test.py
tests/optim/sgd_test.py