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.

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

Investigate using JAX #30

Open
J08nY opened this issue Mar 15, 2023 · 0 comments
Open

Investigate using JAX #30

J08nY opened this issue Mar 15, 2023 · 0 comments
Labels
enhancement New feature or request performance Something is slow
Projects
GPU algorithm implementations

Comments

@J08nY
Copy link
Owner

J08nY commented Mar 15, 2023

JAX while focused on machine learning is a versatile tool for computing on GPUs that has a numpy-compatible API. We could investigate using it for trace operations on GPUs.
@J08nY J08nY added enhancement New feature or request performance Something is slow labels Mar 15, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request performance Something is slow
Projects
GPU algorithm implementations
Status: To do
GPU algorithm implementations
  
To do
Milestone
No milestone
Development

No branches or pull requests
1 participant
@J08nY