Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Native support for Convolution Layer #147

Open
TheodoreEhrenborg opened this issue Sep 17, 2021 · 2 comments
Open

Native support for Convolution Layer #147

TheodoreEhrenborg opened this issue Sep 17, 2021 · 2 comments

Comments

@TheodoreEhrenborg
Copy link

Use case: I’ve written a module in Python that plays a board game. Part of the module uses a convolutional neural network, which I made with TensorFlow Keras.

I’d like to translate my module into Rust to make it run faster. So I need to be able to run a convolutional neural network in Rust. It’s much more convenient for me to run it on a CPU than on a GPU, even though that makes it slower.

Right now Juice doesn’t support running a Convolution Layer on a CPU:

currently the native backend has an incomplete implementation of the Convolution Layer, as such it will panic on use.

So I’d like to request support for this.

I would write it myself, except that I’m an absolute beginner at Rust.

It doesn’t have to be very fast—presumably TensorFlow’s convolutional layer is also slower on a CPU, but that has been fast enough for me.

I will pay $20 (at least—it’s negotiable) to whoever implements this.

(Another option is for me to translate almost all of my module into Rust, except for a Python script that the rest of the module calls (using PyO3?) in order to run the neural network. This seems hacky.)

@TheodoreEhrenborg
Copy link
Author

Oh, this issue seems related.

@drahnr
Copy link
Member

drahnr commented Sep 22, 2021

Hey, yeah, this has been outstanding, I have all the math ready on paper, I just did not get around to impl - and it has been a while.

The code referenced from that other issue unfortunately never appeared in any repo or branch. @subversive-owl unfortunately doesn't seem to be active much anymore on github (correct me if I am wrong!).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants