UT3Tangent.to_jax#

t3toolbox.uniform_manifold.UT3Tangent.to_jax()#

def to_jax(self) -> 'UT3Tangent':

Copy with frame and variation supercores converted to jax arrays (masks stay host numpy).