3 people f1d21afe88 Somewhat more generic broadcasting (#105) ...

* Somewhat more generic broadcasting

* Add TODO

* Set Torch to deterministic in test

Co-authored-by: holonomicjl <[email protected]>

2020-11-11 20:33:00 -08:00

..

test_mnist.py

GPU MaxPool2D.backward(); TinyConvNet train passes (#103)

2020-11-11 07:58:43 -08:00

test_net_speed.py

GPU MaxPool2D.backward(); TinyConvNet train passes (#103)

2020-11-11 07:58:43 -08:00

test_nn.py

unit test for batchnorm2d

2020-10-30 08:19:58 -07:00

test_ops.py

Somewhat more generic broadcasting (#105)

2020-11-11 20:33:00 -08:00

test_optim.py

efficient version of adam (#20)

2020-10-27 15:54:40 -07:00

test_tensor.py

Add GPU Support! (do not merge yet) (#41)

2020-11-01 07:00:49 -08:00