jafioti / luminal

Deep learning at the speed of light.

Home Page:https://luminalai.com

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Convolution

jafioti opened this issue · comments

The convolution operation is important for vision networks and will enable more ops like CumSum and ARange natively.

I belive it can be done like tinygrad does: https://github.com/tinygrad/tinygrad/blob/0fb1d47aa0eb7152073dfbd568893a018cfc4139/tinygrad/tensor.py#L558

This requires #2 to be completed first

Semi-complete. Will close when yolo is finished