`jaxlie.manifold._tree_utils`

Module Contents

Functions

normalize_all(pytree)

Call .normalize() on each Lie group instance in a pytree.

Attributes

`TangentPytree`
`PytreeType`

jaxlie.manifold._tree_utils.TangentPytree

jaxlie.manifold._tree_utils.PytreeType

jaxlie.manifold._tree_utils.normalize_all(pytree)[source]

Call .normalize() on each Lie group instance in a pytree.

Results in a naive projection of each group instance to its respective manifold.

Parameters:: pytree (PytreeType) –
Return type:: PytreeType