jaxlie documentation

Build status icon Mypy status icon Lint status icon Test coverage status icon

jaxlie is a Lie theory library for rigid body transformations and optimization in JAX.

Inheritance diagram of jaxlie.SO2, jaxlie.SO3, jaxlie.SE2, jaxlie.SE3

Current functionality:

  • SO(2), SE(2), SO(3), and SE(3) Lie groups implemented as high-level dataclasses.

  • exp(), log(), adjoint(), multiply(), inverse(), and identity() implementations for each Lie group.

  • Pytree registration for all dataclasses.

  • Helpers + analytical Jacobians for tangent-space optimization (jaxlie.manifold).

Source code on Github.