Bend icon indicating copy to clipboard operation
Bend copied to clipboard

(Request) Autograd / Neural network library

Open j6k4m8 opened this issue 1 year ago • 4 comments

Wondering if anyone on the Bend team or in the community has started thinking about a neural network library that runs on Bend or more generally on the HVM. Is the Bend API mature enough for that right now? Interested in writing one if no one else has started yet :)

j6k4m8 avatar May 20 '24 12:05 j6k4m8

Right now our priority is in finishing more basic things. For a library like that we first need to finish the package manager and finish implementing IO.

Also, traditional neural networks are designed to be extremely performant on the GPU (it's broken down into mostly very homogenous matrix operations). So it probably won't be our priority once we start developing libraries and applications, but if someone in the community want to do it, we'd be very happy.

developedby avatar May 20 '24 13:05 developedby

it would be interesting to see if you could kludge in to the bend c codegen some usage of the Enzyme autodiff library and compile the c code with clang

anandijain avatar May 22 '24 00:05 anandijain

I made a simple MLP if you want to mess around with something similar: https://github.com/akaashdash/bend-mlp

akaashdash avatar May 23 '24 06:05 akaashdash