typing

Typing utilities.

cneuromax.fitting.deeplearning.utils.typing.Batched_data_type = jaxtyping.Num[Tensor, 'BS *_'] | tuple[jaxtyping.Num[Tensor, 'BS *_'], ...] | list[jaxtyping.Num[Tensor, 'BS *_']] | dict[str, jaxtyping.Num[Tensor, 'BS *_']] | transformers.tokenization_utils_base.BatchEncoding

Type hint for batched data.

BS: batch size.