Merge branch 'develop' into develop

Author: aabills

docs/source/user_guide/installation/gnu-linux-mac.rst

@@ -103,19 +103,6 @@ The ``pip install "pybamm[jax]"`` command automatically downloads and installs `
 
 PyBaMM's full `conda-forge distribution <index.rst#installation>`_ (``pybamm``) includes ``jax`` and ``jaxlib`` by default.
 
-.. _optional-iree-mlir-support:
-
-Optional - IREE / MLIR support
-~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
-
-Users can install ``iree`` (for MLIR just-in-time compilation) to use for main expression evaluation in the IDAKLU solver. Requires ``jax``.
-
-.. code:: bash
-
-   pip install "pybamm[iree,jax]"
-
-The ``pip install "pybamm[iree,jax]"`` command automatically downloads and installs ``pybamm`` and the compatible versions of ``jax`` and ``iree`` onto your system.
-
 Uninstall PyBaMM
 ----------------
 

docs/source/user_guide/installation/index.rst

@@ -33,10 +33,9 @@ It can be installed using ``pip`` or ``conda``, or from source.
 Optional solvers
 ----------------
 
-The following solvers are optionally available:
+The following solver is optionally available:
 
 *  `jax <https://jax.readthedocs.io/en/latest/notebooks/quickstart.html>`_ -based solver, see :ref:`optional-jaxsolver` .
-*  `IREE <https://iree.dev/>`_ (`MLIR <https://mlir.llvm.org/>`_) support, see :ref:`optional-iree-mlir-support`.
 
 Dependencies
 ------------
@@ -219,17 +218,6 @@ Dependency                                                                Minimu
 `jaxlib <https://pypi.org/project/jaxlib/>`__                             0.4.20             jax                Support library for JAX
 ========================================================================= ================== ================== =======================
 
-IREE dependencies
-^^^^^^^^^^^^^^^^^
-
-Installable with ``pip install "pybamm[iree]"`` (requires ``jax`` dependencies to be installed).
-
-========================================================================= ================== ================== =======================
-Dependency                                                                Minimum Version    pip extra          Notes
-========================================================================= ================== ================== =======================
-`iree-compiler <https://iree.dev/>`__                                     20240507.886       iree               IREE compiler
-========================================================================= ================== ================== =======================
-
 Full installation guide
 -----------------------
 

pyproject.toml

@@ -115,13 +115,12 @@ dev = [
 
 ]
 # For the Jax solver.
-# Note: These must be kept in sync with the versions defined in pybamm/util.py, and
-#       must remain compatible with IREE (see noxfile.py for IREE compatibility).
+# Note: These must be kept in sync with the versions defined in pybamm/util.py
 jax = [
     "jax==0.4.27",
     "jaxlib==0.4.27",
 ]
-# Contains all optional dependencies, except for jax, iree, and dev dependencies
+# Contains all optional dependencies, except for jax, and dev dependencies
 all = [
     "scikit-fem>=8.1.0",
     "pybamm[examples,plot,cite,bpx,tqdm]",