t3toolbox.backend.common

Attributes

has_jax
has_jax
NDArray
NDArray
is_ndarray
is_ndarray
jax_scan
jax_map
jax_scan

Functions

is_boolean_ndarray(x)
is_boolean_ndarray(x)
ragged_scan(→ Tuple[CarryType, Tuple[Tuple[NDArray, ...)	Similar to jax.lax.scan, except for ragged-sized arrays
numpy_scan(→ Tuple[CarryType, Tuple[NDArray, Ellipsis]])	Similar to jax.lax.scan, except returns numpy arrays instead of jax arrays.
ragged_map(→ Tuple[Tuple[NDArray, Ellipsis], Ellipsis])
numpy_map(→ Tuple[NDArray, Ellipsis])
get_backend(is_uniform, use_jax)
randn(*args, use_jax)

Module Contents

t3toolbox.backend.common.has_jax = False

t3toolbox.backend.common.has_jax = True

t3toolbox.backend.common.NDArray

t3toolbox.backend.common.NDArray

t3toolbox.backend.common.is_ndarray

t3toolbox.backend.common.is_ndarray

t3toolbox.backend.common.jax_scan

t3toolbox.backend.common.jax_map

t3toolbox.backend.common.jax_scan