geohot 17bf90dbe4 unbroadcasting works on the GPU

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

..

test_mnist.py

Topological sort, zero_grads (#119)

2020-11-15 20:25:29 -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

unbroadcasting works on the GPU

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

test_optim.py

efficient version of adam (#20)

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

test_tensor.py

move gradcheck to extra, clean up unbroadcast

2020-11-16 08:03:31 -08:00