unilab.algos.mlx.common.rotation

Functions

axis_angle_to_quat(axis, angle)

Convert MLX axis-angle batches to quaternions.

quat_mul(q1, q2)

Multiply two MLX quaternion batches.

unilab.algos.mlx.common.rotation.quat_mul(q1, q2)[source]

Multiply two MLX quaternion batches.

unilab.algos.mlx.common.rotation.axis_angle_to_quat(axis, angle)[source]

Convert MLX axis-angle batches to quaternions.