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

Add parameter list support to JAX solver (permitting multithreading/GPU execution) #3121

Merged
merged 12 commits into from
Sep 19, 2023
Merged

Add parameter list support to JAX solver (permitting multithreading/GPU execution) #3121

merged 12 commits into from
Sep 19, 2023

Commits on Jul 7, 2023

  1. Allow parameter lists to Jax solver

    @jsbrittain
    jsbrittain committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    f3eaad8 View commit details
    Browse the repository at this point in the history

  2. Use Jax (experimental) sparse matrices

    @jsbrittain
    jsbrittain committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    01c6ed2 View commit details
    Browse the repository at this point in the history

  3. Add parameter list solver to JAX with multithreaded and gpu support

    @jsbrittain
    jsbrittain committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    a15b435 View commit details
    Browse the repository at this point in the history

  4. style: pre-commit fixes

    @pre-commit-ci
    pre-commit-ci[bot] committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    3b9142f View commit details
    Browse the repository at this point in the history

  5. Add changelog entry

    @jsbrittain
    jsbrittain committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    7215685 View commit details
    Browse the repository at this point in the history

  6. Merge branch 'develop' into jax

    @jsbrittain
    jsbrittain committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    093185d View commit details
    Browse the repository at this point in the history

  7. Installer now specifies Jax minor version only

    @jsbrittain
    jsbrittain committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    680a868 View commit details
    Browse the repository at this point in the history

  8. Revert inclusion of experimental jax sparse matrices

    @jsbrittain
    jsbrittain committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    6a75c58 View commit details
    Browse the repository at this point in the history

  9. Permit multiple inputs in Jax unit tests

    @jsbrittain
    jsbrittain committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    efc61be View commit details
    Browse the repository at this point in the history

  10. Only run Jax tests when Jax installed

    @jsbrittain
    jsbrittain committed Jul 7, 2023
    Configuration menu
    Copy the full SHA
    5abff1d View commit details
    Browse the repository at this point in the history

Commits on Aug 1, 2023

  1. Merge branch 'pybamm-team:develop' into jax

    @jsbrittain
    jsbrittain committed Aug 1, 2023
    Configuration menu
    Copy the full SHA
    1badfd0 View commit details
    Browse the repository at this point in the history

Commits on Sep 19, 2023

  1. Merge branch 'develop' into jax

    @jsbrittain
    jsbrittain committed Sep 19, 2023
    Configuration menu
    Copy the full SHA
    8ba70fd View commit details
    Browse the repository at this point in the history