jaxlie.utils

Package Contents

Functions

get_epsilon(dtype)

Helper for grabbing type-specific precision constants.

register_lie_group(*, matrix_dim, parameters_dim, ...)

Decorator for registering Lie group dataclasses.
jaxlie.utils.get_epsilon(dtype)[source]

Helper for grabbing type-specific precision constants.

Parameters:

dtype (jax.numpy.dtype) – Datatype.

Returns:

Output float.

Return type:

float

jaxlie.utils.register_lie_group(*, matrix_dim, parameters_dim, tangent_dim, space_dim)[source]

Decorator for registering Lie group dataclasses.

Sets dimensionality class variables, and marks all methods for JIT compilation.

Parameters:

  • matrix_dim (int) –

  • parameters_dim (int) –

  • tangent_dim (int) –

  • space_dim (int) –

Return type:

Callable[[Type[T]], Type[T]]