jaxsim 0.2.1.dev80__py3-none-any.whl → 0.2.1.dev98__py3-none-any.whl

jaxsim/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.1.dev80'
-__version_tuple__ = version_tuple = (0, 2, 1, 'dev80')
+__version__ = version = '0.2.1.dev98'
+__version_tuple__ = version_tuple = (0, 2, 1, 'dev98')

jaxsim/api/data.py

@@ -30,7 +30,7 @@ except ImportError:
 @jax_dataclasses.pytree_dataclass
 class JaxSimModelData(common.ModelDataWithVelocityRepresentation):
     """
-    Class containing the state of a `JaxSimModel` object.
+    Class containing the data of a `JaxSimModel` object.
     """
 
     state: ODEState
@@ -43,6 +43,24 @@ class JaxSimModelData(common.ModelDataWithVelocityRepresentation):
         default_factory=lambda: jnp.array(0, dtype=jnp.uint64)
     )
 
+    def __hash__(self) -> int:
+
+        return hash(
+            (
+                hash(self.state),
+                hash(tuple(self.gravity.flatten().tolist())),
+                hash(self.soft_contacts_params),
+                hash(jnp.atleast_1d(self.time_ns).flatten().tolist()),
+            )
+        )
+
+    def __eq__(self, other: JaxSimModelData) -> bool:
+
+        if not isinstance(other, JaxSimModelData):
+            return False
+
+        return hash(self) == hash(other)
+
     def valid(self, model: js.model.JaxSimModel | None = None) -> bool:
         """
         Check if the current state is valid for the given model.

jaxsim/api/frame.py

@@ -30,7 +30,7 @@ def idx_of_parent_link(model: js.model.JaxSimModel, *, frame_idx: jtp.IntLike) -
     """
 
     # Get the intermediate representation parsed from the model description.
-    ir = model.description.get()
+    ir = model.description
 
     # Extract the indices of the frame and the link it is attached to.
     F = ir.frames[frame_idx - model.number_of_links()]
@@ -51,7 +51,7 @@ def name_to_idx(model: js.model.JaxSimModel, *, frame_name: str) -> int:
         The index of the frame.
     """
 
-    frame_names = np.array([frame.name for frame in model.description.get().frames])
+    frame_names = np.array([frame.name for frame in model.description.frames])
 
     if frame_name in frame_names:
         idx_in_list = np.argwhere(frame_names == frame_name)
@@ -72,7 +72,7 @@ def idx_to_name(model: js.model.JaxSimModel, *, frame_index: jtp.IntLike) -> str
         The name of the frame.
     """
 
-    return model.description.get().frames[frame_index - model.number_of_links()].name
+    return model.description.frames[frame_index - model.number_of_links()].name
 
 
 @functools.partial(jax.jit, static_argnames=["frame_names"])
@@ -144,7 +144,7 @@ def transform(
     W_H_L = js.link.transform(model=model, data=data, link_index=L)
 
     # Get the static frame pose wrt the parent link.
-    frame = model.description.get().frames[frame_index - model.number_of_links()]
+    frame = model.description.frames[frame_index - model.number_of_links()]
     L_H_F = frame.pose
 
     # Combine the transforms computing the frame pose.

jaxsim/api/kin_dyn_parameters.py

@@ -11,7 +11,7 @@ from jax_dataclasses import Static
 import jaxsim.typing as jtp
 from jaxsim.math import Inertia, JointModel, supported_joint_motion
 from jaxsim.parsers.descriptions import JointDescription, ModelDescription
-from jaxsim.utils import JaxsimDataclass
+from jaxsim.utils import HashedNumpyArray, JaxsimDataclass
 
 
 @jax_dataclasses.pytree_dataclass
@@ -32,8 +32,8 @@ class KynDynParameters(JaxsimDataclass):
 
     # Static
     link_names: Static[tuple[str]]
-    parent_array: Static[jtp.Vector]
-    support_body_array_bool: Static[jtp.Matrix]
+    _parent_array: Static[HashedNumpyArray]
+    _support_body_array_bool: Static[HashedNumpyArray]
 
     # Links
     link_parameters: LinkParameters
@@ -45,6 +45,14 @@ class KynDynParameters(JaxsimDataclass):
     joint_model: JointModel
     joint_parameters: JointParameters | None
 
+    @property
+    def parent_array(self) -> jtp.Vector:
+        return self._parent_array.get()
+
+    @property
+    def support_body_array_bool(self) -> jtp.Matrix:
+        return self._support_body_array_bool.get()
+
     @staticmethod
     def build(model_description: ModelDescription) -> KynDynParameters:
         """
@@ -191,8 +199,8 @@ class KynDynParameters(JaxsimDataclass):
 
         return KynDynParameters(
             link_names=tuple(l.name for l in ordered_links),
-            parent_array=parent_array,
-            support_body_array_bool=support_body_array_bool,
+            _parent_array=HashedNumpyArray(array=parent_array),
+            _support_body_array_bool=HashedNumpyArray(array=support_body_array_bool),
             link_parameters=link_parameters,
             joint_model=joint_model,
             joint_parameters=joint_parameters,
@@ -204,23 +212,18 @@ class KynDynParameters(JaxsimDataclass):
         if not isinstance(other, KynDynParameters):
             return False
 
-        equal = True
-        equal = equal and self.number_of_links() == other.number_of_links()
-        equal = equal and self.number_of_joints() == other.number_of_joints()
-        equal = equal and jnp.allclose(self.parent_array, other.parent_array)
-
-        return equal
+        return hash(self) == hash(other)
 
     def __hash__(self) -> int:
 
-        h = (
-            hash(self.number_of_links()),
-            hash(self.number_of_joints()),
-            hash(tuple(self.parent_array.tolist())),
+        return hash(
+            (
+                hash(self.number_of_links()),
+                hash(self.number_of_joints()),
+                hash(tuple(jnp.atleast_1d(self.parent_array).flatten().tolist())),
+            )
         )
 
-        return hash(h)
-
     # =============================
     # Helpers to extract parameters
     # =============================
@@ -388,7 +391,7 @@ class KynDynParameters(JaxsimDataclass):
             pre_H_suc_J, S_J = jax.vmap(supported_joint_motion)(
                 jnp.array(self.joint_model.joint_types[1:]).astype(int),
                 jnp.array(joint_positions),
-                jnp.array(self.joint_model.joint_axis),
+                jnp.array([j.axis for j in self.joint_model.joint_axis]),
             )
 
         # Extract the transforms and motion subspaces of the joints.

jaxsim/api/model.py

@@ -32,18 +32,37 @@ class JaxSimModel(JaxsimDataclass):
     terrain: Static[jaxsim.terrain.Terrain] = dataclasses.field(
         default=jaxsim.terrain.FlatTerrain(), repr=False, compare=False, hash=False
     )
+    kin_dyn_parameters: js.kin_dyn_parameters.KynDynParameters | None = (
+        dataclasses.field(default=None, repr=False, compare=False, hash=False)
+    )
 
     built_from: Static[str | pathlib.Path | rod.Model | None] = dataclasses.field(
         default=None, repr=False, compare=False, hash=False
     )
 
-    description: Static[
+    _description: Static[
         HashlessObject[jaxsim.parsers.descriptions.ModelDescription | None]
     ] = dataclasses.field(default=None, repr=False, compare=False, hash=False)
 
-    kin_dyn_parameters: js.kin_dyn_parameters.KynDynParameters | None = (
-        dataclasses.field(default=None, repr=False, compare=False, hash=False)
-    )
+    @property
+    def description(self) -> jaxsim.parsers.descriptions.ModelDescription:
+        return self._description.get()
+
+    def __eq__(self, other: JaxSimModel) -> bool:
+
+        if not isinstance(other, JaxSimModel):
+            return False
+
+        return hash(self) == hash(other)
+
+    def __hash__(self) -> int:
+
+        return hash(
+            (
+                hash(self.model_name),
+                hash(self.kin_dyn_parameters),
+            )
+        )
 
     # ========================
     # Initialization and state
@@ -137,7 +156,7 @@ class JaxSimModel(JaxsimDataclass):
         # Build the model
         model = JaxSimModel(
             model_name=model_name,
-            description=HashlessObject(obj=model_description),
+            _description=HashlessObject(obj=model_description),
             kin_dyn_parameters=js.kin_dyn_parameters.KynDynParameters.build(
                 model_description=model_description
             ),
@@ -260,7 +279,7 @@ class JaxSimModel(JaxsimDataclass):
             The names of the links in the model.
         """
 
-        return tuple([frame.name for frame in self.description.get().frames])
+        return tuple(frame.name for frame in self.description.frames)
 
 
 # =====================
@@ -297,7 +316,7 @@ def reduce(
 
     # Copy the model description with a deep copy of the joints.
     intermediate_description = dataclasses.replace(
-        model.description.get(), joints=copy.deepcopy(model.description.get().joints)
+        model.description, joints=copy.deepcopy(model.description.joints)
     )
 
     # Update the initial position of the joints.

jaxsim/api/ode_data.py

@@ -281,6 +281,24 @@ class PhysicsModelState(JaxsimDataclass):
         default_factory=lambda: jnp.zeros(3)
     )
 
+    def __hash__(self) -> int:
+
+        return hash(
+            (
+                hash(tuple(jnp.atleast_1d(self.joint_positions.flatten().tolist()))),
+                hash(tuple(jnp.atleast_1d(self.joint_velocities.flatten().tolist()))),
+                hash(tuple(self.base_position.flatten().tolist())),
+                hash(tuple(self.base_quaternion.flatten().tolist())),
+            )
+        )
+
+    def __eq__(self, other: PhysicsModelState) -> bool:
+
+        if not isinstance(other, PhysicsModelState):
+            return False
+
+        return hash(self) == hash(other)
+
     @staticmethod
     def build_from_jaxsim_model(
         model: js.model.JaxSimModel | None = None,
@@ -593,6 +611,19 @@ class SoftContactsState(JaxsimDataclass):
 
     tangential_deformation: jtp.Matrix
 
+    def __hash__(self) -> int:
+
+        return hash(
+            tuple(jnp.atleast_1d(self.tangential_deformation.flatten()).tolist())
+        )
+
+    def __eq__(self, other: SoftContactsState) -> bool:
+
+        if not isinstance(other, SoftContactsState):
+            return False
+
+        return hash(self) == hash(other)
+
     @staticmethod
     def build_from_jaxsim_model(
         model: js.model.JaxSimModel | None = None,

jaxsim/math/joint_model.py

@@ -39,7 +39,7 @@ class JointModel:
 
     joint_dofs: Static[tuple[int, ...]]
     joint_names: Static[tuple[str, ...]]
-    joint_types: Static[tuple[JointType, ...]]
+    joint_types: Static[tuple[int, ...]]
     joint_axis: Static[tuple[JointGenericAxis, ...]]
 
     @staticmethod
@@ -109,7 +109,7 @@ class JointModel:
             joint_dofs=tuple([base_dofs] + [int(1) for _ in ordered_joints]),
             joint_names=tuple(["world_to_base"] + [j.name for j in ordered_joints]),
             joint_types=tuple([JointType.Fixed] + [j.jtype for j in ordered_joints]),
-            joint_axis=tuple([j.axis for j in ordered_joints]),
+            joint_axis=tuple(JointGenericAxis(axis=j.axis) for j in ordered_joints),
         )
 
     def parent_H_child(
@@ -201,7 +201,7 @@ class JointModel:
         pre_H_suc, S = supported_joint_motion(
             self.joint_types[joint_index],
             joint_position,
-            self.joint_axis[joint_index],
+            self.joint_axis[joint_index].axis,
         )
 
         return pre_H_suc, S
@@ -224,9 +224,9 @@ class JointModel:
 
 @jax.jit
 def supported_joint_motion(
-    joint_type: JointType,
+    joint_type: jtp.IntLike,
     joint_position: jtp.VectorLike,
-    joint_axis: JointGenericAxis,
+    joint_axis: jtp.VectorLike | None = None,
     /,
 ) -> tuple[jtp.Matrix, jtp.Array]:
     """
@@ -234,8 +234,8 @@ def supported_joint_motion(
 
     Args:
         joint_type: The type of the joint.
-        joint_axis: The axis of rotation or translation of the joint.
         joint_position: The position of the joint.
+        joint_axis: The optional 3D axis of rotation or translation of the joint.
 
     Returns:
         A tuple containing the homogeneous transformation and the motion subspace.
@@ -244,26 +244,33 @@ def supported_joint_motion(
     # Prepare the joint position
     s = jnp.array(joint_position).astype(float)
 
-    def compute_F():
+    def compute_F() -> tuple[jtp.Matrix, jtp.Array]:
         return jaxlie.SE3.identity(), jnp.zeros(shape=(6, 1))
 
-    def compute_R():
+    def compute_R() -> tuple[jtp.Matrix, jtp.Array]:
+
+        # Get the additional argument specifying the joint axis.
+        # This is a metadata required by only some joint types.
+        axis = jnp.array(joint_axis).astype(float).squeeze()
+
         pre_H_suc = jaxlie.SE3.from_rotation(
-            rotation=jaxlie.SO3.from_matrix(
-                Rotation.from_axis_angle(vector=s * joint_axis)
-            )
+            rotation=jaxlie.SO3.from_matrix(Rotation.from_axis_angle(vector=s * axis))
         )
 
-        S = jnp.vstack(jnp.hstack([jnp.zeros(3), joint_axis.squeeze()]))
+        S = jnp.vstack(jnp.hstack([jnp.zeros(3), axis]))
+
         return pre_H_suc, S
 
-    def compute_P():
-        pre_H_suc = jaxlie.SE3.from_rotation_and_translation(
-            rotation=jaxlie.SO3.identity(),
-            translation=jnp.array(s * joint_axis),
-        )
+    def compute_P() -> tuple[jtp.Matrix, jtp.Array]:
+
+        # Get the additional argument specifying the joint axis.
+        # This is a metadata required by only some joint types.
+        axis = jnp.array(joint_axis).astype(float).squeeze()
+
+        pre_H_suc = jaxlie.SE3.from_translation(translation=jnp.array(s * axis))
+
+        S = jnp.vstack(jnp.hstack([axis, jnp.zeros(3)]))
 
-        S = jnp.vstack(jnp.hstack([joint_axis.squeeze(), jnp.zeros(3)]))
         return pre_H_suc, S
 
     pre_H_suc, S = jax.lax.switch(

jaxsim/parsers/descriptions/joint.py

@@ -30,9 +30,11 @@ class JointGenericAxis:
     axis: jtp.Vector
 
     def __hash__(self) -> int:
-        return hash((tuple(np.array(self.axis).tolist())))
+
+        return hash(tuple(self.axis.tolist()))
 
     def __eq__(self, other: JointGenericAxis) -> bool:
+
         if not isinstance(other, JointGenericAxis):
             return False
 

jaxsim/rbda/soft_contacts.py

@@ -29,6 +29,23 @@ class SoftContactsParams(JaxsimDataclass):
         default_factory=lambda: jnp.array(0.5, dtype=float)
     )
 
+    def __hash__(self) -> int:
+
+        return hash(
+            (
+                hash(tuple(jnp.atleast_1d(self.K).flatten().tolist())),
+                hash(tuple(jnp.atleast_1d(self.D).flatten().tolist())),
+                hash(tuple(jnp.atleast_1d(self.mu).flatten().tolist())),
+            )
+        )
+
+    def __eq__(self, other: SoftContactsParams) -> bool:
+
+        if not isinstance(other, SoftContactsParams):
+            return NotImplemented
+
+        return hash(self) == hash(other)
+
     @staticmethod
     def build(
         K: jtp.FloatLike = 1e6, D: jtp.FloatLike = 2_000, mu: jtp.FloatLike = 0.5

jaxsim/utils/__init__.py

@@ -1,5 +1,5 @@
 from jax_dataclasses._copy_and_mutate import _Mutability as Mutability
 
-from .hashless import HashlessObject
 from .jaxsim_dataclass import JaxsimDataclass
 from .tracing import not_tracing, tracing
+from .wrappers import HashedNumpyArray, HashlessObject

jaxsim/utils/wrappers.py

@@ -0,0 +1,78 @@
+from __future__ import annotations
+
+import dataclasses
+from typing import Generic, TypeVar
+
+import jax
+import jax_dataclasses
+import numpy as np
+import numpy.typing as npt
+
+T = TypeVar("T")
+
+
+@dataclasses.dataclass
+class HashlessObject(Generic[T]):
+    """
+    A class that wraps an object and makes it hashless.
+
+    This is useful for creating particular JAX pytrees.
+    For example, to create a pytree with a static leaf that is ignored
+    by JAX when it compares two instances to trigger a JIT recompilation.
+    """
+
+    obj: T
+
+    def get(self: HashlessObject[T]) -> T:
+        return self.obj
+
+    def __hash__(self) -> int:
+
+        return 0
+
+    def __eq__(self, other: HashlessObject[T]) -> bool:
+
+        if not isinstance(other, HashlessObject) and isinstance(
+            other.get(), type(self.get())
+        ):
+            return False
+
+        return hash(self) == hash(other)
+
+
+@jax_dataclasses.pytree_dataclass
+class HashedNumpyArray:
+    """
+    A class that wraps a numpy array and makes it hashable.
+
+    This is useful for creating particular JAX pytrees.
+    For example, to create a pytree with a plain NumPy or JAX NumPy array as static leaf.
+
+    Note:
+        Calculating with the wrapper class the hash of a very large array can be
+        very expensive. If the array is large and only the equality operator is needed,
+        set `large_array=True` to use a faster comparison method.
+    """
+
+    array: jax.Array | npt.NDArray
+
+    large_array: jax_dataclasses.Static[bool] = dataclasses.field(
+        default=False, repr=False, compare=False, hash=False
+    )
+
+    def get(self) -> jax.Array | npt.NDArray:
+        return self.array
+
+    def __hash__(self) -> int:
+
+        return hash(tuple(np.atleast_1d(self.array).flatten().tolist()))
+
+    def __eq__(self, other: HashedNumpyArray) -> bool:
+
+        if not isinstance(other, HashedNumpyArray):
+            return False
+
+        if self.large_array:
+            return np.array_equal(self.array, other.array)
+
+        return hash(self) == hash(other)

{jaxsim-0.2.1.dev80.dist-info → jaxsim-0.2.1.dev98.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jaxsim
-Version: 0.2.1.dev80
+Version: 0.2.1.dev98
 Home-page: https://github.com/ami-iit/jaxsim
 Author: Diego Ferigo
 Author-email: diego.ferigo@iit.it

{jaxsim-0.2.1.dev80.dist-info → jaxsim-0.2.1.dev98.dist-info}/RECORD

@@ -1,19 +1,19 @@
 jaxsim/__init__.py,sha256=OcrfoYS1DGcmAGqu2AqlCTiUVxcpi-IsVwcr_16x74Q,1789
-jaxsim/_version.py,sha256=6bZIeO5gVEXbcUw14r0OEvahs0AZBEq5fAmJ16fr2fU,426
+jaxsim/_version.py,sha256=PlgYXFRQrTcDBWrHgW3TWsyD0WOKMdyNQ1dtp2gm-oU,426
 jaxsim/logging.py,sha256=c4zhwBKf9eAYAHVp62kTEllqdsZgh0K-kPKVy8L3elU,1584
 jaxsim/typing.py,sha256=MeuOCQtLAr-sPkvB_sU8FtwGNRirz1auCwIgRC-QZl8,646
 jaxsim/api/__init__.py,sha256=8eV22t2S3UwNyCg8karPetG1dmX1VDBXkyv28_FwNQA,210
 jaxsim/api/com.py,sha256=Yof6otFi-mLWAs1rqjmeNJTOWIH9gn7BdU5EIjiL6Ts,13481
 jaxsim/api/common.py,sha256=DV-WZG28sikXopNv458aYvpLjmiAtFr5LRscOwXusuk,6640
 jaxsim/api/contact.py,sha256=Cvr-EfQtHP3nymtWdo-9WWU24Bkta-2Pp3nKsdjo6uc,12778
-jaxsim/api/data.py,sha256=GXSCfq4_PsRZuUwsmcl18e2Ppdtu4ykYs0oWih8J2ZI,26775
-jaxsim/api/frame.py,sha256=68liCmwFDikuP0GQOpo-nF7aVeHJ173e1USSOwskTP0,6246
+jaxsim/api/data.py,sha256=xfKJz6Rw0YTk-EHCGiT8BFQrs_ggOz01lRi1Qh1mb28,27256
+jaxsim/api/frame.py,sha256=0YXOrGmx3cSQqa4_Ky-n6zyup3I3xvXNEgub-Bc5xUw,6222
 jaxsim/api/joint.py,sha256=-5DogPg4g4mmLckyVIVNjwv-Rxz0IWS7_md9nDlhPWA,4581
-jaxsim/api/kin_dyn_parameters.py,sha256=9xdHS3lDWSEdV0qAqCVoS6cSMUw17ebPJdbBfq4vjVU,25959
+jaxsim/api/kin_dyn_parameters.py,sha256=zMca7OmCsCWK_cavLTSZSeYh9Qu1-409cdsyWvWPAUQ,26090
 jaxsim/api/link.py,sha256=rypTwkMf9HJ5UuAtHRJh0LqqdJWcLKTtTjWcjduEsF0,9842
-jaxsim/api/model.py,sha256=7qBsneoc3dWNIsd6INzItGaIWSTgO98TQUZaeyVRrdk,53809
+jaxsim/api/model.py,sha256=Wwg3Wp9jm2Ah7wjvzou7oZYdZk2iTlfzSidp6GNwfJ0,54263
 jaxsim/api/ode.py,sha256=6l-6i2YHagsQvR8Ac-_fmO6P0hBVT6NkHhwXnrdITEg,9785
-jaxsim/api/ode_data.py,sha256=dwRFVPJ30XMmdUbPXEu7YxsQ97jZP4L4fd5ZzhrO5Ys,22184
+jaxsim/api/ode_data.py,sha256=D6FzMkvY_qNuoFEImyp7sxAk-0pJOd3oZeSr9bBTcLk,23089
 jaxsim/api/references.py,sha256=Lvskf17r619KKxwCJP7hAAty2kaXgDXJX1uKqoDIDgo,15483
 jaxsim/integrators/__init__.py,sha256=hxvOD-VK_mmd6v31wtC-nb28AYve1gLuZCNLV9wS-Kg,103
 jaxsim/integrators/common.py,sha256=9HXRVFo95Mpt6RcVhBrOfvOO7mDxqbkXeg_lKUibEFY,20693
@@ -23,7 +23,7 @@ jaxsim/math/__init__.py,sha256=inJ9nRFkqstuGa8OyFkfWVudo5U9Ug4WgDBuKva8AIA,337
 jaxsim/math/adjoint.py,sha256=DT21izjVW497GRrgNfx8tv0ZeWW5QncWMGMhI0acUNw,4425
 jaxsim/math/cross.py,sha256=U7yEx_l75mSy5g6O-jsjBztApvxC3WaV4MpkS5tThu4,1330
 jaxsim/math/inertia.py,sha256=UAB7ym4gXFanejcs_ovZMpteHCc6poWYmt-mLmd5hhk,1640
-jaxsim/math/joint_model.py,sha256=L_rLVUONJclAaPgZmtNrZRB6X7tx6g5xZoVH2E2PU7I,9535
+jaxsim/math/joint_model.py,sha256=lGZxwuGqIXYeF2dYC5I248-mRNJLvb86iomq7yLEBmE,9909
 jaxsim/math/quaternion.py,sha256=X9b8jHf0QemKUjIZSnXRJc3DdMr42CBhBy_mi9_X_AM,5068
 jaxsim/math/rotation.py,sha256=Z90daUjGpuNEVLfWB3SVtM9EtwAIaneVj9A9UpWXqhA,2182
 jaxsim/math/skew.py,sha256=oOGSSR8PUGROl6IJFlrmu6K3gPH-u16hUPfKIkcVv9o,1177
@@ -37,7 +37,7 @@ jaxsim/parsers/__init__.py,sha256=sonYi-bBWAoB04kp1mxT4uIORxjb7SdZ0ukGPmVx98Y,44
 jaxsim/parsers/kinematic_graph.py,sha256=zFt7x7pPGJar36Azukdi1eI_sa1kMWD3B8kZqcHx6iw,33934
 jaxsim/parsers/descriptions/__init__.py,sha256=PbIlunVfb59pB5jSX97YVpMAANRZPRkJ0X-hS14rzv4,221
 jaxsim/parsers/descriptions/collision.py,sha256=HUWwuRgI9KznY29FFw1_zU3bGigDEezrcPOJSxSJGNU,3382
-jaxsim/parsers/descriptions/joint.py,sha256=r7teUWJyw62WS7qNCB_9FGQD1hvGHBepk-2LCSlmcSc,3052
+jaxsim/parsers/descriptions/joint.py,sha256=lRnYMmjpASpz0Ueuqzwnj5Ze4yLRgPTx66H0_kbQnNI,3042
 jaxsim/parsers/descriptions/link.py,sha256=GC-6ZgRZuRVpcRo1sY6YaR8lkCHkR4DvHNs2Ydw_tn4,2887
 jaxsim/parsers/descriptions/model.py,sha256=uO5xOJtViihVPnSSsmfQJvCh45ANyi9KYAzLOhH0R8g,8993
 jaxsim/parsers/rod/__init__.py,sha256=G2vqlLajBLUc4gyzXwsEI2Wsi4TMOIF9bLDFeT6KrGU,92
@@ -50,16 +50,16 @@ jaxsim/rbda/crba.py,sha256=GodskOZjtrSlbQAqxRv1un_706O7BaJK-U2qa18vJk8,4741
 jaxsim/rbda/forward_kinematics.py,sha256=OHugNU7C0UxYAW0o1rqH1ZgniSwurz6L1T1MJxfxq08,3418
 jaxsim/rbda/jacobian.py,sha256=9LGGy9ya5m5U0mBmV1NFH5XYZpEMYbx74qnYBvZs7Ok,6360
 jaxsim/rbda/rnea.py,sha256=DjwkvXQVUSUclM3Uy3UPZ2tao91R5dGd4o7TsS2qObI,7650
-jaxsim/rbda/soft_contacts.py,sha256=2EZ9Lw4nFWqXTMEeYsirl17H61s82SmTZllKVsP1Yek,10759
+jaxsim/rbda/soft_contacts.py,sha256=52zJOF31hFpqoaOednTvi8j_UxhRcdGNjzOPb2v2MPc,11257
 jaxsim/rbda/utils.py,sha256=zpbFM2Iq8cntku0BFVu9nfEqZhInCWi9D2INT6MFEI8,5003
 jaxsim/terrain/__init__.py,sha256=f7lVX-iNpH_wkkjef9Qpjh19TTAUOQw76EiLYJDVizc,78
 jaxsim/terrain/terrain.py,sha256=q0xkWqEShVq-p1j2abTLZq8sEhjyJwquxQKm80PaHhM,2161
-jaxsim/utils/__init__.py,sha256=tnQq1_CavdfeKaLYt3pmO7Jk4MU2RwwQU_qICkjyoTY,197
-jaxsim/utils/hashless.py,sha256=bFIwKeo9KiWwsY8QM55duEGGQOyyJ4jQyPcuqTLEp5k,297
+jaxsim/utils/__init__.py,sha256=Y5zyoRevl3EMVQadhZ4EtSwTEkDt2vcnFoRhPJjKTZ0,215
 jaxsim/utils/jaxsim_dataclass.py,sha256=h26timZ_XrBL_Q_oymv-DkQd-EcUiHn8QexAaZXBY9c,11396
 jaxsim/utils/tracing.py,sha256=KDMoyVPlu2NJvFkhtZwq5AkqMMgajt3munvJom-vEjQ,650
-jaxsim-0.2.1.dev80.dist-info/LICENSE,sha256=eaYdFmdeMbiIoIiPzEK0MjP1S9wtFXjXNR5er49uLR0,1546
-jaxsim-0.2.1.dev80.dist-info/METADATA,sha256=yFYgYOpZwSa-j1g4zj-4mp2B3z4TkL_9kbF-Rd5Q3hI,9744
-jaxsim-0.2.1.dev80.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-jaxsim-0.2.1.dev80.dist-info/top_level.txt,sha256=LxGMA8FLtXjQ6oI7N5gd_R_oSUHxpXxUEOfT1xS_ni0,7
-jaxsim-0.2.1.dev80.dist-info/RECORD,,
+jaxsim/utils/wrappers.py,sha256=EJMcblYKUjxw9HJShVf81Ig3pHUJno6Dx6h-RnY--wM,2040
+jaxsim-0.2.1.dev98.dist-info/LICENSE,sha256=eaYdFmdeMbiIoIiPzEK0MjP1S9wtFXjXNR5er49uLR0,1546
+jaxsim-0.2.1.dev98.dist-info/METADATA,sha256=_aAfq6LooqjTyZY6utw1p0NrkwVZr10NC9Gw-MqCR1Y,9744
+jaxsim-0.2.1.dev98.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+jaxsim-0.2.1.dev98.dist-info/top_level.txt,sha256=LxGMA8FLtXjQ6oI7N5gd_R_oSUHxpXxUEOfT1xS_ni0,7
+jaxsim-0.2.1.dev98.dist-info/RECORD,,

jaxsim/utils/hashless.py

@@ -1,18 +0,0 @@
-from __future__ import annotations
-
-import dataclasses
-from typing import Generic, TypeVar
-
-T = TypeVar("T")
-
-
-@dataclasses.dataclass
-class HashlessObject(Generic[T]):
-
-    obj: T
-
-    def get(self: HashlessObject[T]) -> T:
-        return self.obj
-
-    def __hash__(self) -> int:
-        return 0