probly.representation.sample

Classes representing prediction samples.

Submodules

array	Numpy-based sample representation.
array_functions	Numpy array function implementations for sample arrays.
axis_tracking	Implementation of axis tracking through NumPy-style indexing operations.
jax	JAX sample implementation.
jax_axis_tracking	Axis-tracking for PyTorch tensors.
torch	Torch sample implementation.
torch_axis_tracking	Axis-tracking for PyTorch tensors.
torch_functions	Torch function implementations for sample tensors.