* add nn ops * fix style * small * conv * Testing * add docstring * fix typo * add more testing
* add nn ops * fix style * move conv out