baybe.utils.torch

Torch utilities shipped as separate module for lazy-loading.

Module Attributes

VARNAME_TORCH_USE_SINGLE_PRECISION	Environment variable name for enforcing single precision in torch.
DTypeFloatTorch	Floating point data type used for torch tensors.