-
Notifications
You must be signed in to change notification settings - Fork 60
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
GPU support? #1
Comments
@ahundt indeed, great suggestion. We are investigating extensions and GPU (PyTorch) support is definitely one of them; will keep you posted! |
any progress on gpu support? |
Look at this issue. |
Have you solved the problem about running on GPU?could you share? @fabbrimatteo |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
This looks very interesting but I don't have TPU access. Do you have any plans to tweak the code to also support running on GPU/CPU?
The text was updated successfully, but these errors were encountered: