geohot 58e703d099 fix tests

2020-11-10 09:49:19 -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

some broadcasting, pool test is fail

2020-11-07 11:29:42 -08:00

ops.py

match torch api for pad2d

2020-11-09 17:48:56 -08:00

opsgpu.py

fix tests

2020-11-10 09:49:19 -08:00

optim.py

gpu SGD runs now

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

tensor.py

refactor profiler

2020-11-10 07:31:16 -08:00

utils.py

deleted unnecessary import in utils (#78)

2020-11-08 15:55:16 -08:00