Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion exp/2025/08/27/inverse-segement/src/20-forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
2 changes: 1 addition & 1 deletion exp/2025/08/27/inverse-segement/src/30-inverse.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
2 changes: 1 addition & 1 deletion exp/2025/08/27/inverse/src/20-forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
2 changes: 1 addition & 1 deletion exp/2025/08/27/inverse/src/30-inverse.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion exp/2025/09/10/inverse-mixed/src/20-forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/10/inverse-mixed/src/30-inverse.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,21 +6,21 @@
import equinox as eqx
import jax
import jax.numpy as jnp
import liblaf.apple.warp.sim as sim_wp
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)
Expand Down
2 changes: 1 addition & 1 deletion exp/2025/09/10/inverse/src/20-forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/10/inverse/src/30-inverse.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,21 +5,21 @@
import equinox as eqx
import jax
import jax.numpy as jnp
import liblaf.apple.warp.sim as sim_wp
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)
Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/10/inverse/src/31-inverse-v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,21 +6,21 @@
import equinox as eqx
import jax
import jax.numpy as jnp
import liblaf.apple.warp.sim as sim_wp
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)
Expand Down
2 changes: 1 addition & 1 deletion exp/2025/09/24/inverse-grin/src/20-forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down
2 changes: 1 addition & 1 deletion exp/2025/09/24/inverse-grin/src/21-forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/24/inverse-grin/src/30-inverse.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 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"

Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/24/inverse-grin/src/31-inverse.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 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"

Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/24/inverse-grin/src/32-inverse.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 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"

Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/24/inverse-grin/src/33-inverse-no-reg.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 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"

Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/24/inverse-grin/src/34-inverse-no-reg.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 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"

Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/24/inverse-grin/src/35-inverse-small-reg.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 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"

Expand Down
8 changes: 4 additions & 4 deletions exp/2025/09/24/inverse-grin/src/36-inverse-abs.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 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"

Expand Down
Loading
Loading