geohot ec03eb44bd tinygrad does forward pass convs on GPU

2020-11-07 10:15:56 -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

weee, logsoftmax on GPU. very slow

2020-11-07 07:59:37 -08:00

opsgpu.py

tinygrad does forward pass convs on GPU

2020-11-07 10:15:56 -08:00

optim.py

gpu SGD runs now

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

tensor.py

yayay test_sgd_gpu passes

2020-11-07 08:48:17 -08:00

utils.py

fetch() 'ran out of input' fix (#65)

2020-11-07 09:45:10 -08:00