George Hotz 237bbe3625 sigmoid on GPU

2020-11-07 07:23:46 -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

factor out a couple nn ops

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

ops.py

sigmoid on GPU

2020-11-07 07:23:46 -08:00

opsgpu.py

sigmoid on GPU

2020-11-07 07:23:46 -08:00

optim.py

gpu SGD runs now

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

tensor.py

make OpenCL work elsewhere

2020-11-05 07:59:55 -08:00

utils.py

Fix temp location for windows (#39)

2020-10-31 17:21:14 -07:00