UT3Variations.to_jax#

t3toolbox.uniform_frame_variations_format.UT3Variations.to_jax()#

def to_jax(self) -> 'UT3Variations':
    # Supercores -> jax; masks stay host numpy (a jax mask is a tracer under jit). See docs/uniform_*.