jaxlie.types

Package Contents

jaxlie.types.Matrix

Type alias for jnp.ndarray. Should not be instantiated.

Refers to a square matrix, typically with shape (Group.matrix_dim, Group.matrix_dim). For adjoints, shape should be (Group.tangent_dim, Group.tangent_dim).

jaxlie.types.Vector

Type alias for jnp.ndarray. Should not be instantiated.

Refers to a general 1D array.

jaxlie.types.TangentVector

Type alias for jnp.ndarray. Should not be instantiated.

Refers to a 1D array with shape (Group.tangent_dim,).