NeuralLinkandGitHub 9f77fd6135 🔨 refactor optim (#156) ...

* 🔨 refactor optim

* 🔨 refactor optim

* 🔨 more clean up

2020-12-08 14:16:31 -08:00

..

__init__.py

factor out a couple nn ops

2020-10-29 08:01:12 -07:00

nn.py

batchnorm work

2020-12-06 14:40:07 -08:00

ops_cpu.py

Support scalars by kartik4949

2020-12-08 09:52:07 -08:00

ops_gpu.py

Support scalars by kartik4949

2020-12-08 09:52:07 -08:00

optim.py

🔨 refactor optim (#156)

2020-12-08 14:16:31 -08:00

tensor.py

fix rsub

2020-12-08 10:05:21 -08:00

utils.py

replace layer_init_uniform with .uniform

2020-12-06 13:44:31 -08:00