Name		Name	Last commit message	Last commit date
README.md		README.md
batch_norm_1d.py		batch_norm_1d.py
conv_1d.py		conv_1d.py
gru_cell.py		gru_cell.py
layer_norm.py		layer_norm.py
linear.py		linear.py
lstm_cell.py		lstm_cell.py
lstm_cell_obj.py		lstm_cell_obj.py
max_pool_1d.py		max_pool_1d.py
multihead_attention.py		multihead_attention.py
rnn.py		rnn.py
rnn_cell.py		rnn_cell.py
rnn_cell_obj.py		rnn_cell_obj.py
transformer.py		transformer.py

Repository files navigation

n2

Simple implementations of torch.nn modules.

Motivation: To provide simplified pytorch.nn compatible modules that are easy to read and understand.

TODO:

BatchNorm2d MaxPool2d Conv2d

About

Simple implementations of torch.nn modules

2 stars

0 watching

Report repository

Releases

Packages

Contributors

Languages