George Hotz 0e6832a8ea support torch GPU, only autoinit cuda in the buffer

2022-01-15 19:15:12 -08:00

..

ops

support torch GPU, only autoinit cuda in the buffer

2022-01-15 19:15:12 -08:00

__init__.py

tinygrad.utils -> extra.utils

2020-12-12 15:26:07 -08:00

helpers.py

factor out binary_broadcast

2021-11-29 03:49:12 -05:00

nn.py

clean up Conv2d

2021-11-30 11:02:55 -05:00

optim.py

minor cleanups, yolo work

2021-01-03 08:14:16 -08:00

tensor.py

save two lines

2021-11-30 01:16:48 -05:00