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.

Sign up for GitHub

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

Jump to bottom

GPU porting for pure Python #20

Open

ArnauMiro opened this issue Oct 25, 2023 · 0 comments

Assignees

Labels

Owner

ArnauMiro commented Oct 25, 2023

As stated by @b-fg, the CuPy library could be used for GPU porting. This library is interesting for the following main points in my opinion:

Linear algebra libraries are implemented.
Distributed access through NCCL.
Profiling through NSYS.
Low level memory management for the future.

ArnauMiro added the enhancement label

ArnauMiro assigned ArnauMiro and bef-18

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