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

Refactor to JAX? #1

Open

JoFrhwld opened this issue Nov 6, 2023 · 1 comment

Member

JoFrhwld commented Nov 6, 2023

JAX is a project from Google designed to have as similar an API as numpy, but can be run on either a CPU or a GPU.

Possible benefits

Faster processing
GPU capability could entail some future-proofing
Similarity to numpy should reduce development friction

Possible downsides

Google project get killed sometimes
There are "sharp bits"

Member Author

JoFrhwld commented Nov 11, 2023

I don't think this is ready for primetime. There are a few too many system specific installment requirements and undocumented dependency mismatches

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