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

..

__init__.py

factor out a couple nn ops

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

gradcheck.py

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

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

nn.py

promote swish to a tensor ops

2020-11-10 15:48:11 -08:00

ops.py

touch ups

2020-11-10 15:44:47 -08:00

opsgpu.py

Somewhat more generic broadcasting (#105)

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

optim.py

gpu SGD runs now

2020-11-02 07:19:12 -08:00

tensor.py

clean up imports

2020-11-10 15:53:05 -08:00

utils.py

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

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