opentau.utils

Utility functions and helpers for the OpenTau package.

This module provides various utility functions for logging, random number generation, training utilities, device management, and other common operations used throughout the OpenTau codebase.

Modules

accelerate_utils

Utilities for managing Accelerate accelerator instances.

benchmark

Utilities for benchmarking and timing code execution.

fake_tensor

Utilities for working with PyTorch FakeTensor.

hub

Utilities for interacting with the Hugging Face Hub.

import_utils

Utilities for checking package availability and versions.

io_utils

Utilities for file I/O operations.

libero_dataset_recorder

Utilities for recording and converting LIBERO datasets.

logging_utils

Utilities for tracking and logging training metrics.

monkey_patch

Monkey patches to inject behaviour into computational graph construction.

random_utils

Utilities for managing random number generator states.

ros2lerobot

ROS 2 to LeRobot dataset extractors and utilities.

train_utils

Utilities for training checkpoint management and state persistence.

transformers_patch

Module for patching transformers

utils

General utility functions for device management, logging, and common operations.

vision_utils

Vision-tower helpers reused across policies (pi06, pi07, ...).