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

Use JAX sparse module for numerical impedance predictions #596

Open
BradyPlanden opened this issue Dec 13, 2024 · 0 comments
Open

Use JAX sparse module for numerical impedance predictions #596

BradyPlanden opened this issue Dec 13, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@BradyPlanden
Copy link
Member

Feature description

Add a Jax sparse solving method for the numeric impedance prediction to enable gradient-based optimisers.

Motivation

Currently, our implementation of numerical impedance uses the numpy sparse solver spsolve. Jax has experimental functionality for sparse matrices and operations (see here) with functionality for jax.grad and similar functions.

Possible implementation

No response

Additional context

No response
@BradyPlanden BradyPlanden added the enhancement New feature or request label Dec 13, 2024
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
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@BradyPlanden