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

Adds JAX IDAKLU solver integration #481

Open
wants to merge 22 commits into
base: develop
Choose a base branch
from
Open

Adds JAX IDAKLU solver integration #481

wants to merge 22 commits into from

Commits on Aug 20, 2024

  1. working commit, adds [jax] optional dependency to pybamm, example ida… 

    …klu-jax solver
    @BradyPlanden
    BradyPlanden committed Aug 20, 2024
    Configuration menu
    Copy the full SHA
    1e08d6e View commit details
    Browse the repository at this point in the history

  2. adds experimental sub-directory, sync commit

    @BradyPlanden
    BradyPlanden committed Aug 20, 2024
    Configuration menu
    Copy the full SHA
    79234f9 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'refs/heads/develop' into jaxify-idaklu-implementation

    @BradyPlanden
    BradyPlanden committed Aug 20, 2024
    Configuration menu
    Copy the full SHA
    7d8aa9c View commit details
    Browse the repository at this point in the history

Commits on Aug 28, 2024

  1. Merge branch 'refs/heads/develop' into jaxify-idaklu-implementation

    @BradyPlanden
    BradyPlanden committed Aug 28, 2024
    Configuration menu
    Copy the full SHA
    ab861a9 View commit details
    Browse the repository at this point in the history

Commits on Aug 31, 2024

  1. feat: adds Jax functionality via idaklu_jax solver, with example and … 

    …benchmarks. Adds JaxSumSquaredError and JaxLogNormalLikelihood.
    @BradyPlanden
    BradyPlanden committed Aug 31, 2024
    Configuration menu
    Copy the full SHA
    f2bf143 View commit details
    Browse the repository at this point in the history

Commits on Sep 2, 2024

  1. Merge branch 'refs/heads/develop' into jaxify-idaklu-implementation 

    # Conflicts:
#	pyproject.toml
    @BradyPlanden
    BradyPlanden committed Sep 2, 2024
    Configuration menu
    Copy the full SHA
    025c7d8 View commit details
    Browse the repository at this point in the history

  2. Post merge fixes, update base_model solver property

    @BradyPlanden
    BradyPlanden committed Sep 2, 2024
    Configuration menu
    Copy the full SHA
    d3a18d5 View commit details
    Browse the repository at this point in the history

  3. examples: update benchmarking script

    @BradyPlanden
    BradyPlanden committed Sep 2, 2024
    Configuration menu
    Copy the full SHA
    eb08c00 View commit details
    Browse the repository at this point in the history

Commits on Sep 6, 2024

  1. Merge branch 'develop' into jaxify-idaklu-implementation

    @BradyPlanden
    BradyPlanden authored Sep 6, 2024
    Configuration menu
    Copy the full SHA
    92265b6 View commit details
    Browse the repository at this point in the history

Commits on Sep 24, 2024

  1. Merge branch 'refs/heads/develop' into jaxify-idaklu-implementation 

    # Conflicts:
#	pyproject.toml
    @BradyPlanden
    BradyPlanden committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    f1d03a9 View commit details
    Browse the repository at this point in the history

  2. updts docstrings, jax arg to models, adds tests.

    @BradyPlanden
    BradyPlanden committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    3ecd5dd View commit details
    Browse the repository at this point in the history

Commits on Sep 25, 2024

  1. add coverage, refactor BaseModel solver setter

    @BradyPlanden
    BradyPlanden committed Sep 25, 2024
    Configuration menu
    Copy the full SHA
    9794910 View commit details
    Browse the repository at this point in the history

  2. adds coverage

    @BradyPlanden
    BradyPlanden committed Sep 25, 2024
    Configuration menu
    Copy the full SHA
    df1ba0c View commit details
    Browse the repository at this point in the history

  3. convert BaseModel.calculate_sensitivites to property

    @BradyPlanden
    BradyPlanden committed Sep 25, 2024
    Configuration menu
    Copy the full SHA
    d770315 View commit details
    Browse the repository at this point in the history

  4. add changelog entry

    @BradyPlanden
    BradyPlanden committed Sep 25, 2024
    Configuration menu
    Copy the full SHA
    daae4ba View commit details
    Browse the repository at this point in the history

Commits on Sep 30, 2024

  1. feat: removes Jax arg from BaseModel, BaseProblem.model is copied ins… 

    …tance
    @BradyPlanden
    BradyPlanden committed Sep 30, 2024
    Configuration menu
    Copy the full SHA
    424699b View commit details
    Browse the repository at this point in the history

Commits on Oct 11, 2024

  1. Merge branch 'develop' into jaxify-idaklu-implementation

    @BradyPlanden
    BradyPlanden authored Oct 11, 2024
    Configuration menu
    Copy the full SHA
    2695fd7 View commit details
    Browse the repository at this point in the history

Commits on Oct 21, 2024

  1. fix: cost shape for CMAES, adds example comments for jax solver

    @BradyPlanden
    BradyPlanden committed Oct 21, 2024
    Configuration menu
    Copy the full SHA
    23b8e3d View commit details
    Browse the repository at this point in the history

  2. Merge branch 'develop' into jaxify-idaklu-implementation 

    # Conflicts:
#	CHANGELOG.md
#	examples/notebooks/1-single-pulse-circuit-model.ipynb
    @BradyPlanden
    BradyPlanden committed Oct 21, 2024
    Configuration menu
    Copy the full SHA
    bba2e56 View commit details
    Browse the repository at this point in the history

  3. fix: update model reparameterisation as model is now copied within Ba… 

    …seProblem()
    @BradyPlanden
    BradyPlanden committed Oct 21, 2024
    Configuration menu
    Copy the full SHA
    f7672ce View commit details
    Browse the repository at this point in the history

Commits on Oct 23, 2024

  1. Merge branch 'develop' into jaxify-idaklu-implementation 

    # Conflicts:
#	examples/notebooks/single_pulse_circuit_model.ipynb
    @BradyPlanden
    BradyPlanden committed Oct 23, 2024
    Configuration menu
    Copy the full SHA
    0549d14 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'develop' into jaxify-idaklu-implementation 

    # Conflicts:
#	examples/notebooks/single_pulse_circuit_model.ipynb
    @BradyPlanden
    BradyPlanden committed Oct 23, 2024
    Configuration menu
    Copy the full SHA
    44e3095 View commit details
    Browse the repository at this point in the history