From bb2efc219be4ea645bc4bf8a12511b0a5a1012f7 Mon Sep 17 00:00:00 2001
From: "renovate[bot]" <29139614+renovate[bot]@users.noreply.github.com>
Date: Thu, 26 Feb 2026 21:51:10 +0000
Subject: [PATCH 1/2] chore(deps): update github artifact actions

---
 .github/workflows/mega-linter.yaml | 2 +-
 .github/workflows/release.yaml     | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/mega-linter.yaml b/.github/workflows/mega-linter.yaml
index 8704a1ac..1e544b8d 100644
--- a/.github/workflows/mega-linter.yaml
+++ b/.github/workflows/mega-linter.yaml
@@ -43,7 +43,7 @@ jobs:
           token: ${{ steps.auth.outputs.token }}
       - if: success() || failure()
         name: Upload Reports
-        uses: actions/upload-artifact@v6
+        uses: actions/upload-artifact@v7
         with:
           name: mega-linter-reports
           path: |-
diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
index 0a608193..2f852978 100644
--- a/.github/workflows/release.yaml
+++ b/.github/workflows/release.yaml
@@ -45,7 +45,7 @@ jobs:
       url: https://pypi.org/project/${{ needs.build.outputs.package-name }}/${{ needs.build.outputs.package-version }}/
     steps:
       - name: Download Artifacts
-        uses: actions/download-artifact@v7
+        uses: actions/download-artifact@v8
         with:
           name: ${{ needs.build.outputs.artifact-name }}
           path: dist
@@ -65,7 +65,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Download Artifacts
-        uses: actions/download-artifact@v7
+        uses: actions/download-artifact@v8
         with:
           name: ${{ needs.build.outputs.artifact-name }}
           path: dist

From addab3905c5891237a9b895a9c5898ca118e88db Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Thu, 26 Feb 2026 21:51:23 +0000
Subject: [PATCH 2/2] chore(ci): auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 exp/2025/08/27/inverse-segement/src/20-forward.py      |  2 +-
 exp/2025/08/27/inverse-segement/src/30-inverse.py      |  2 +-
 exp/2025/08/27/inverse/src/20-forward.py               |  2 +-
 exp/2025/08/27/inverse/src/30-inverse.py               |  2 +-
 exp/2025/09/10/inverse-mixed/src/20-forward.py         |  2 +-
 exp/2025/09/10/inverse-mixed/src/30-inverse.py         | 10 +++++-----
 exp/2025/09/10/inverse/src/20-forward.py               |  2 +-
 exp/2025/09/10/inverse/src/30-inverse.py               | 10 +++++-----
 exp/2025/09/10/inverse/src/31-inverse-v2.py            | 10 +++++-----
 exp/2025/09/24/inverse-grin/src/20-forward.py          |  2 +-
 exp/2025/09/24/inverse-grin/src/21-forward.py          |  2 +-
 exp/2025/09/24/inverse-grin/src/30-inverse.py          | 10 +++++-----
 exp/2025/09/24/inverse-grin/src/31-inverse.py          | 10 +++++-----
 exp/2025/09/24/inverse-grin/src/32-inverse.py          | 10 +++++-----
 exp/2025/09/24/inverse-grin/src/33-inverse-no-reg.py   | 10 +++++-----
 exp/2025/09/24/inverse-grin/src/34-inverse-no-reg.py   | 10 +++++-----
 .../09/24/inverse-grin/src/35-inverse-small-reg.py     | 10 +++++-----
 exp/2025/09/24/inverse-grin/src/36-inverse-abs.py      | 10 +++++-----
 .../09/24/inverse-grin/src/37-inverse-quadratic.py     | 10 +++++-----
 exp/2025/09/24/inverse-grin/src/38-inverse-cbrt.py     | 10 +++++-----
 exp/2025/09/24/inverse-grin/src/39-inverse-sparse.py   | 10 +++++-----
 exp/2025/09/24/inverse-grin/src/41-inspect-inverse.py  |  2 +-
 exp/2025/10/22/inverse-flame/src/30-animate.py         |  2 +-
 exp/2025/10/29/muscle-fiber-direction/src/10-solve.py  |  2 +-
 tests/warp/energies/elastic/hyperelastic/common.py     |  4 ++--
 .../elastic/hyperelastic/func/test_gradients.py        |  2 +-
 .../elastic/hyperelastic/func/test_hess_diag.py        |  4 ++--
 .../elastic/hyperelastic/func/test_hess_prod.py        |  4 ++--
 .../elastic/hyperelastic/func/test_hess_quad.py        |  4 ++--
 tests/warp/energies/elastic/hyperelastic/test_arap.py  |  2 +-
 .../energies/elastic/hyperelastic/test_arap_muscle.py  |  2 +-
 tests/warp/energies/elastic/hyperelastic/test_phace.py |  2 +-
 32 files changed, 88 insertions(+), 88 deletions(-)

diff --git a/exp/2025/08/27/inverse-segement/src/20-forward.py b/exp/2025/08/27/inverse-segement/src/20-forward.py
index 2ae0e88b..f8c087f0 100644
--- a/exp/2025/08/27/inverse-segement/src/20-forward.py
+++ b/exp/2025/08/27/inverse-segement/src/20-forward.py
@@ -4,13 +4,13 @@
 import jax.numpy as jnp
 import numpy as np
 import pyvista as pv
+from liblaf.apple.jax.typing import Vector
 from loguru import logger
 
 from liblaf import cherries, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim
 from liblaf.apple.jax import sim as sim_jax
-from liblaf.apple.jax.typing import Vector
 
 jax.config.update("jax_enable_x64", True)  # noqa: FBT003
 
diff --git a/exp/2025/08/27/inverse-segement/src/30-inverse.py b/exp/2025/08/27/inverse-segement/src/30-inverse.py
index 598abe1f..f5d64b4d 100644
--- a/exp/2025/08/27/inverse-segement/src/30-inverse.py
+++ b/exp/2025/08/27/inverse-segement/src/30-inverse.py
@@ -9,13 +9,13 @@
 import numpy as np
 import pyvista as pv
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
 from loguru import logger
 
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
 from liblaf.apple.jax import sim as sim_jax
-from liblaf.apple.jax.typing import Scalar, Vector
 
 
 class Config(cherries.BaseConfig):
diff --git a/exp/2025/08/27/inverse/src/20-forward.py b/exp/2025/08/27/inverse/src/20-forward.py
index e14058b2..ee9df9ba 100644
--- a/exp/2025/08/27/inverse/src/20-forward.py
+++ b/exp/2025/08/27/inverse/src/20-forward.py
@@ -4,13 +4,13 @@
 import jax.numpy as jnp
 import numpy as np
 import pyvista as pv
+from liblaf.apple.jax.typing import Vector
 from loguru import logger
 
 from liblaf import cherries, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim
 from liblaf.apple.jax import sim as sim_jax
-from liblaf.apple.jax.typing import Vector
 
 jax.config.update("jax_enable_x64", True)  # noqa: FBT003
 
diff --git a/exp/2025/08/27/inverse/src/30-inverse.py b/exp/2025/08/27/inverse/src/30-inverse.py
index 4eab8679..e2690e02 100644
--- a/exp/2025/08/27/inverse/src/30-inverse.py
+++ b/exp/2025/08/27/inverse/src/30-inverse.py
@@ -9,13 +9,13 @@
 import numpy as np
 import pyvista as pv
 from jaxtyping import Array, Float
+from liblaf.apple.jax.typing import Scalar, Vector
 from loguru import logger
 
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
 from liblaf.apple.jax import sim as sim_jax
-from liblaf.apple.jax.typing import Scalar, Vector
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 jax.config.update("jax_enable_x64", True)  # noqa: FBT003
diff --git a/exp/2025/09/10/inverse-mixed/src/20-forward.py b/exp/2025/09/10/inverse-mixed/src/20-forward.py
index b9782b3b..587dacf4 100644
--- a/exp/2025/09/10/inverse-mixed/src/20-forward.py
+++ b/exp/2025/09/10/inverse-mixed/src/20-forward.py
@@ -3,12 +3,12 @@
 import jax.numpy as jnp
 import numpy as np
 import pyvista as pv
+from liblaf.apple.jax.typing import Vector
 from loguru import logger
 
 from liblaf import cherries, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim
-from liblaf.apple.jax.typing import Vector
 from liblaf.apple.warp import sim as sim_wp
 
 
diff --git a/exp/2025/09/10/inverse-mixed/src/30-inverse.py b/exp/2025/09/10/inverse-mixed/src/30-inverse.py
index 04df21db..f1928eaa 100644
--- a/exp/2025/09/10/inverse-mixed/src/30-inverse.py
+++ b/exp/2025/09/10/inverse-mixed/src/30-inverse.py
@@ -6,21 +6,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.sim.energy.elastic import utils
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.sim.energy.elastic import utils
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 
 @lx.is_positive_semidefinite.register(lx.DiagonalLinearOperator)
diff --git a/exp/2025/09/10/inverse/src/20-forward.py b/exp/2025/09/10/inverse/src/20-forward.py
index affcfc69..8d16e075 100644
--- a/exp/2025/09/10/inverse/src/20-forward.py
+++ b/exp/2025/09/10/inverse/src/20-forward.py
@@ -3,12 +3,12 @@
 import jax.numpy as jnp
 import numpy as np
 import pyvista as pv
+from liblaf.apple.jax.typing import Vector
 from loguru import logger
 
 from liblaf import cherries, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim
-from liblaf.apple.jax.typing import Vector
 from liblaf.apple.warp import sim as sim_wp
 
 
diff --git a/exp/2025/09/10/inverse/src/30-inverse.py b/exp/2025/09/10/inverse/src/30-inverse.py
index b7b926cd..59947d5d 100644
--- a/exp/2025/09/10/inverse/src/30-inverse.py
+++ b/exp/2025/09/10/inverse/src/30-inverse.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Float
+from liblaf.apple.jax.sim.energy.elastic import utils
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.sim.energy.elastic import utils
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 
 @lx.is_positive_semidefinite.register(lx.DiagonalLinearOperator)
diff --git a/exp/2025/09/10/inverse/src/31-inverse-v2.py b/exp/2025/09/10/inverse/src/31-inverse-v2.py
index aa97b326..8b57a256 100644
--- a/exp/2025/09/10/inverse/src/31-inverse-v2.py
+++ b/exp/2025/09/10/inverse/src/31-inverse-v2.py
@@ -6,21 +6,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.sim.energy.elastic import utils
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.sim.energy.elastic import utils
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 
 @lx.is_positive_semidefinite.register(lx.DiagonalLinearOperator)
diff --git a/exp/2025/09/24/inverse-grin/src/20-forward.py b/exp/2025/09/24/inverse-grin/src/20-forward.py
index f4d8e179..06eaf357 100644
--- a/exp/2025/09/24/inverse-grin/src/20-forward.py
+++ b/exp/2025/09/24/inverse-grin/src/20-forward.py
@@ -3,12 +3,12 @@
 import jax.numpy as jnp
 import numpy as np
 import pyvista as pv
+from liblaf.apple.jax.typing import Vector
 from loguru import logger
 
 from liblaf import cherries, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim
-from liblaf.apple.jax.typing import Vector
 from liblaf.apple.warp import sim as sim_wp
 
 
diff --git a/exp/2025/09/24/inverse-grin/src/21-forward.py b/exp/2025/09/24/inverse-grin/src/21-forward.py
index f601ed2b..dce79e61 100644
--- a/exp/2025/09/24/inverse-grin/src/21-forward.py
+++ b/exp/2025/09/24/inverse-grin/src/21-forward.py
@@ -3,12 +3,12 @@
 import jax.numpy as jnp
 import numpy as np
 import pyvista as pv
+from liblaf.apple.jax.typing import Vector
 from liblaf.peach import optim
 from loguru import logger
 
 from liblaf import cherries, melon
 from liblaf.apple import sim
-from liblaf.apple.jax.typing import Vector
 from liblaf.apple.warp import sim as sim_wp
 
 
diff --git a/exp/2025/09/24/inverse-grin/src/30-inverse.py b/exp/2025/09/24/inverse-grin/src/30-inverse.py
index fe10e27a..c6065d6b 100644
--- a/exp/2025/09/24/inverse-grin/src/30-inverse.py
+++ b/exp/2025/09/24/inverse-grin/src/30-inverse.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/31-inverse.py b/exp/2025/09/24/inverse-grin/src/31-inverse.py
index 8ae3671f..51457515 100644
--- a/exp/2025/09/24/inverse-grin/src/31-inverse.py
+++ b/exp/2025/09/24/inverse-grin/src/31-inverse.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/32-inverse.py b/exp/2025/09/24/inverse-grin/src/32-inverse.py
index d1e15c78..8b1dfb76 100644
--- a/exp/2025/09/24/inverse-grin/src/32-inverse.py
+++ b/exp/2025/09/24/inverse-grin/src/32-inverse.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/33-inverse-no-reg.py b/exp/2025/09/24/inverse-grin/src/33-inverse-no-reg.py
index 30d39a25..76e0f560 100644
--- a/exp/2025/09/24/inverse-grin/src/33-inverse-no-reg.py
+++ b/exp/2025/09/24/inverse-grin/src/33-inverse-no-reg.py
@@ -5,22 +5,22 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import scipy.optimize
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/34-inverse-no-reg.py b/exp/2025/09/24/inverse-grin/src/34-inverse-no-reg.py
index d8cfd962..1872f228 100644
--- a/exp/2025/09/24/inverse-grin/src/34-inverse-no-reg.py
+++ b/exp/2025/09/24/inverse-grin/src/34-inverse-no-reg.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/35-inverse-small-reg.py b/exp/2025/09/24/inverse-grin/src/35-inverse-small-reg.py
index 39468d46..3659b870 100644
--- a/exp/2025/09/24/inverse-grin/src/35-inverse-small-reg.py
+++ b/exp/2025/09/24/inverse-grin/src/35-inverse-small-reg.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/36-inverse-abs.py b/exp/2025/09/24/inverse-grin/src/36-inverse-abs.py
index d9418a6d..ade7b814 100644
--- a/exp/2025/09/24/inverse-grin/src/36-inverse-abs.py
+++ b/exp/2025/09/24/inverse-grin/src/36-inverse-abs.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/37-inverse-quadratic.py b/exp/2025/09/24/inverse-grin/src/37-inverse-quadratic.py
index abf5cfe4..e6757dbf 100644
--- a/exp/2025/09/24/inverse-grin/src/37-inverse-quadratic.py
+++ b/exp/2025/09/24/inverse-grin/src/37-inverse-quadratic.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/38-inverse-cbrt.py b/exp/2025/09/24/inverse-grin/src/38-inverse-cbrt.py
index 8e757e2c..ba2c8900 100644
--- a/exp/2025/09/24/inverse-grin/src/38-inverse-cbrt.py
+++ b/exp/2025/09/24/inverse-grin/src/38-inverse-cbrt.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/39-inverse-sparse.py b/exp/2025/09/24/inverse-grin/src/39-inverse-sparse.py
index a3a5b8f1..57d57742 100644
--- a/exp/2025/09/24/inverse-grin/src/39-inverse-sparse.py
+++ b/exp/2025/09/24/inverse-grin/src/39-inverse-sparse.py
@@ -5,21 +5,21 @@
 import equinox as eqx
 import jax
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
+import liblaf.apple.warp.sim as sim_wp
+import liblaf.apple.warp.utils as wp_utils
 import lineax as lx
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Bool, Float
+from liblaf.apple.jax.typing import Scalar, Vector
+from liblaf.apple.warp.typing import vec6
 from loguru import logger
 
-import liblaf.apple.jax.sim as sim_jax
-import liblaf.apple.warp.sim as sim_wp
-import liblaf.apple.warp.utils as wp_utils
 from liblaf import cherries, grapes, melon
 from liblaf.apple import sim
 from liblaf.apple.jax import optim, tree
-from liblaf.apple.jax.typing import Scalar, Vector
-from liblaf.apple.warp.typing import vec6
 
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = "0.25"
 
diff --git a/exp/2025/09/24/inverse-grin/src/41-inspect-inverse.py b/exp/2025/09/24/inverse-grin/src/41-inspect-inverse.py
index 18f8a305..ed52391d 100644
--- a/exp/2025/09/24/inverse-grin/src/41-inspect-inverse.py
+++ b/exp/2025/09/24/inverse-grin/src/41-inspect-inverse.py
@@ -1,9 +1,9 @@
 import jax.numpy as jnp
+import liblaf.apple.jax.sim as sim_jax
 import numpy as np
 import pyvista as pv
 from jaxtyping import Array, Float
 
-import liblaf.apple.jax.sim as sim_jax
 from liblaf import melon
 
 
diff --git a/exp/2025/10/22/inverse-flame/src/30-animate.py b/exp/2025/10/22/inverse-flame/src/30-animate.py
index 7e1613ea..9d8f0048 100644
--- a/exp/2025/10/22/inverse-flame/src/30-animate.py
+++ b/exp/2025/10/22/inverse-flame/src/30-animate.py
@@ -6,6 +6,7 @@
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, Float
+from liblaf.apple.warp.typing import vec6
 from liblaf.peach import tree
 from liblaf.peach.optim import PNCG, Objective
 
@@ -13,7 +14,6 @@
 from liblaf.apple import sim
 from liblaf.apple.warp import sim as sim_wp
 from liblaf.apple.warp import utils as wpu
-from liblaf.apple.warp.typing import vec6
 
 logger: logging.Logger = logging.getLogger(__name__)
 
diff --git a/exp/2025/10/29/muscle-fiber-direction/src/10-solve.py b/exp/2025/10/29/muscle-fiber-direction/src/10-solve.py
index f1a2ed47..11d24ee1 100644
--- a/exp/2025/10/29/muscle-fiber-direction/src/10-solve.py
+++ b/exp/2025/10/29/muscle-fiber-direction/src/10-solve.py
@@ -3,12 +3,12 @@
 import pyvista as pv
 from jax import Array
 from jaxtyping import Integer
+from liblaf.apple.jax.typing import Scalar, Vector
 from liblaf.peach import optim
 from liblaf.peach import tree_utils as tree
 
 from liblaf import grapes, melon
 from liblaf.apple.jax import sim as sim_jax
-from liblaf.apple.jax.typing import Scalar, Vector
 
 
 @tree.define
diff --git a/tests/warp/energies/elastic/hyperelastic/common.py b/tests/warp/energies/elastic/hyperelastic/common.py
index 65f1da1d..c5ee72f5 100644
--- a/tests/warp/energies/elastic/hyperelastic/common.py
+++ b/tests/warp/energies/elastic/hyperelastic/common.py
@@ -2,15 +2,15 @@
 
 import jax
 import jax.numpy as jnp
+import liblaf.apple.warp.utils as wpu
 import numpy as np
 import pyvista as pv
 import warp as wp
 from jaxtyping import Array, ArrayLike, Float, Key
+from liblaf.apple.warp.energies.elastic.hyperelastic import Hyperelastic
 
-import liblaf.apple.warp.utils as wpu
 from liblaf.apple import Model
 from liblaf.apple.jax import testing
-from liblaf.apple.warp.energies.elastic.hyperelastic import Hyperelastic
 
 type EnergyParams = Mapping[str, Array]
 type Full = Float[Array, "points 3"]
diff --git a/tests/warp/energies/elastic/hyperelastic/func/test_gradients.py b/tests/warp/energies/elastic/hyperelastic/func/test_gradients.py
index 910da93e..95dbf9e1 100644
--- a/tests/warp/energies/elastic/hyperelastic/func/test_gradients.py
+++ b/tests/warp/energies/elastic/hyperelastic/func/test_gradients.py
@@ -5,11 +5,11 @@
 import jax.test_util
 import warp as wp
 from jaxtyping import Array, Float
+from liblaf.apple.warp.energies.elastic.hyperelastic import func
 
 from liblaf.apple.jax import testing
 from liblaf.apple.warp import math
 from liblaf.apple.warp import utils as wpu
-from liblaf.apple.warp.energies.elastic.hyperelastic import func
 
 
 class IdentityFunction(Protocol):
diff --git a/tests/warp/energies/elastic/hyperelastic/func/test_hess_diag.py b/tests/warp/energies/elastic/hyperelastic/func/test_hess_diag.py
index 1d9c408d..6418bec0 100644
--- a/tests/warp/energies/elastic/hyperelastic/func/test_hess_diag.py
+++ b/tests/warp/energies/elastic/hyperelastic/func/test_hess_diag.py
@@ -2,13 +2,13 @@
 
 import hypothesis
 import jax.numpy as jnp
+import liblaf.apple.warp.utils as wpu
 import numpy as np
 import warp as wp
 from jaxtyping import Array, DTypeLike, Float
+from liblaf.apple.warp.energies.elastic.hyperelastic import func
 
-import liblaf.apple.warp.utils as wpu
 from liblaf.apple.jax import testing
-from liblaf.apple.warp.energies.elastic.hyperelastic import func
 
 
 def diag_from_quad(
diff --git a/tests/warp/energies/elastic/hyperelastic/func/test_hess_prod.py b/tests/warp/energies/elastic/hyperelastic/func/test_hess_prod.py
index 0fb14274..5c9f8d07 100644
--- a/tests/warp/energies/elastic/hyperelastic/func/test_hess_prod.py
+++ b/tests/warp/energies/elastic/hyperelastic/func/test_hess_prod.py
@@ -2,13 +2,13 @@
 
 import einops
 import hypothesis
+import liblaf.apple.warp.utils as wpu
 import numpy as np
 import warp as wp
 from jaxtyping import Array, Float
+from liblaf.apple.warp.energies.elastic.hyperelastic import func
 
-import liblaf.apple.warp.utils as wpu
 from liblaf.apple.jax import testing
-from liblaf.apple.warp.energies.elastic.hyperelastic import func
 
 
 def check_prod(
diff --git a/tests/warp/energies/elastic/hyperelastic/func/test_hess_quad.py b/tests/warp/energies/elastic/hyperelastic/func/test_hess_quad.py
index ed516767..09dd77dc 100644
--- a/tests/warp/energies/elastic/hyperelastic/func/test_hess_quad.py
+++ b/tests/warp/energies/elastic/hyperelastic/func/test_hess_quad.py
@@ -2,14 +2,14 @@
 from typing import Any, no_type_check
 
 import hypothesis
+import liblaf.apple.warp.utils as wpu
 import numpy as np
 import warp as wp
 from jaxtyping import Array, Float
+from liblaf.apple.warp.energies.elastic.hyperelastic import func
 
-import liblaf.apple.warp.utils as wpu
 from liblaf.apple.jax import testing
 from liblaf.apple.warp import math
-from liblaf.apple.warp.energies.elastic.hyperelastic import func
 
 type Mat33 = Float[Array, "batch 3 3"]
 type Mat43 = Float[Array, "batch 4 3"]
diff --git a/tests/warp/energies/elastic/hyperelastic/test_arap.py b/tests/warp/energies/elastic/hyperelastic/test_arap.py
index 024ed2b6..4f4ab5ea 100644
--- a/tests/warp/energies/elastic/hyperelastic/test_arap.py
+++ b/tests/warp/energies/elastic/hyperelastic/test_arap.py
@@ -2,11 +2,11 @@
 import numpy as np
 import pytest
 import pyvista as pv
+from liblaf.apple.warp.energies.elastic.hyperelastic import Arap
 
 from liblaf.apple import Model, ModelBuilder
 from liblaf.apple.consts import MU
 from liblaf.apple.jax import testing
-from liblaf.apple.warp.energies.elastic.hyperelastic import Arap
 
 from . import common
 
diff --git a/tests/warp/energies/elastic/hyperelastic/test_arap_muscle.py b/tests/warp/energies/elastic/hyperelastic/test_arap_muscle.py
index 53a12a40..617afd77 100644
--- a/tests/warp/energies/elastic/hyperelastic/test_arap_muscle.py
+++ b/tests/warp/energies/elastic/hyperelastic/test_arap_muscle.py
@@ -2,11 +2,11 @@
 import numpy as np
 import pytest
 import pyvista as pv
+from liblaf.apple.warp.energies.elastic.hyperelastic import ArapMuscle
 
 from liblaf.apple import Model, ModelBuilder
 from liblaf.apple.consts import ACTIVATION, MU
 from liblaf.apple.jax import testing
-from liblaf.apple.warp.energies.elastic.hyperelastic import ArapMuscle
 
 from . import common
 
diff --git a/tests/warp/energies/elastic/hyperelastic/test_phace.py b/tests/warp/energies/elastic/hyperelastic/test_phace.py
index d21c14e9..ae4e8834 100644
--- a/tests/warp/energies/elastic/hyperelastic/test_phace.py
+++ b/tests/warp/energies/elastic/hyperelastic/test_phace.py
@@ -2,11 +2,11 @@
 import numpy as np
 import pytest
 import pyvista as pv
+from liblaf.apple.warp.energies.elastic.hyperelastic import Phace
 
 from liblaf.apple import Model, ModelBuilder
 from liblaf.apple.consts import ACTIVATION, LAMBDA, MU, MUSCLE_FRACTION
 from liblaf.apple.jax import testing
-from liblaf.apple.warp.energies.elastic.hyperelastic import Phace
 
 from . import common