George Hotz f178d23ff3 gpu relu is good

2020-11-02 08:25:32 -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

ops work

2020-11-02 07:03:23 -08:00

opsgpu.py

gpu relu is good

2020-11-02 08:25:32 -08:00

optim.py

gpu SGD runs now

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

tensor.py

the power of small

2020-11-02 07:36:34 -08:00

utils.py

Fix temp location for windows (#39)

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