Skip to content

mollerhoj/nnn

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits

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

Resources

Readme
Activity

Stars

2 stars

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages