Index

_

__abs__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __add__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __array_namespace__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __array_ufunc__ (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) __call__() (autoray.autoray.Composed method) (autoray.autoray.InjectDtypeDevice method) (autoray.compiler.AutoCompiled method) (autoray.compiler.CompileJax method) (autoray.compiler.CompilePytensor method) (autoray.compiler.CompilePython method) (autoray.compiler.CompileTensorFlow method) (autoray.compiler.CompileTorch method) (autoray.compiler.CompileTorch2 method) (autoray.lazy.core.Function method) (autoray.lazy.Function method) __eq__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __floordiv__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __ge__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __getattribute__() (autoray.autoray.AutoNamespace method) __getitem__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __getstate__() (autoray.lazy.core.Function method) (autoray.lazy.Function method) __gt__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __hash__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __iter__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __le__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __len__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __lt__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __matmul__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __mod__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __mul__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __ne__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __neg__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __pow__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __radd__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __repr__() (autoray.autoray.AutoNamespace method) (autoray.autoray.Composed method) (autoray.autoray.InjectDtypeDevice method) (autoray.autoray.Leaf method) (autoray.lazy.core.Function method) (autoray.lazy.core.LazyArray method) (autoray.lazy.core.Placeholder method) (autoray.lazy.Function method) (autoray.lazy.LazyArray method) __rfloordiv__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __rmatmul__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __rmul__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __rpow__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method)

__rsub__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __rtruediv__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __setstate__() (autoray.lazy.core.Function method) (autoray.lazy.Function method) __slots__ (autoray.autoray.InjectDtypeDevice attribute) (autoray.autoray.Leaf attribute) (autoray.lazy.core.Function attribute) (autoray.lazy.core.LazyArray attribute) (autoray.lazy.core.Placeholder attribute) (autoray.lazy.Function attribute) (autoray.lazy.LazyArray attribute) __sub__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __truediv__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __version__ (in module autoray._version) __version_tuple__ (in module autoray._version) _add_sharing_cache() (in module autoray.lazy.core) _always_the_same() (in module autoray.autoray) _BACKEND_ALIASES (in module autoray.autoray) _backend_lock (in module autoray.autoray) _backend_lookup (in module autoray.compiler) _basic_einsum_parse_input() (in module autoray.lazy.core) _builtin_dtype_lookup (in module autoray.autoray) _choose_backend() (in module autoray.autoray) _compiler_lookup (in module autoray.compiler) _COMPLEX_DTYPES (in module autoray.autoray) _COMPOSED_FUNCTION_GENERATORS (in module autoray.autoray) _CREATION_INJECT (in module autoray.autoray) _CREATION_ROUTINES (in module autoray.autoray) _CUSTOM_BACKENDS (in module autoray.autoray) _CUSTOM_WRAPPERS (in module autoray.autoray) _default_infer_from_sig() (in module autoray.autoray) _default_infer_from_sig_threadaware() (in module autoray.autoray) _DISPATCHERS (in module autoray.autoray) _DOUBLE_DTYPES (in module autoray.autoray) _DTYPE_MAP (in module autoray.autoray) _DTYPES_COMPLEX_EQUIV (in module autoray.lazy.core) _DTYPES_REAL_EQUIV (in module autoray.lazy.core) _EMPTY_DICT (in module autoray.lazy.core) _find_common_backend_cached() (in module autoray.lazy.core) _find_common_dtype() (in module autoray.lazy.core) _FUNC_ALIASES (in module autoray.autoray) _FUNCS (in module autoray.autoray) _get_fn() (autoray.autoray.AutoNamespace method) _get_parse_einsum_input() (in module autoray.lazy.core) _get_py_shape() (in module autoray.lazy.core) _get_submodule() (autoray.autoray.AutoNamespace method) _global_backend (in module autoray.autoray) _global_backends_threadaware (in module autoray.autoray) _handle_size_to_shape() (in module autoray.autoray) _importing_thrid (in module autoray.autoray) _infer_auto (in module autoray.autoray) _infer_class_backend_cached() (in module autoray.autoray) _infer_class_backend_multi_cached() (in module autoray.autoray) _inferrer_global (in module autoray.autoray) _inferrers_threadaware (in module autoray.autoray) _JAX_RANDOM_KEY (in module autoray.autoray) _materialize() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) _materialize_dispatch (in module autoray.lazy.core) _MODULE_ALIASES (in module autoray.autoray) _paddle_dtype_name_conversion (in module autoray.autoray) _remove_sharing_cache() (in module autoray.lazy.core) _reshape_tuple() (in module autoray.lazy.core) _SHARING_STACK (in module autoray.lazy.core) _stringify_dispatch (in module autoray.lazy.core) _SUBMODULE_ALIASES (in module autoray.autoray) _torch_get_dtype_name() (in module autoray.autoray) _torch_reduce_translation (in module autoray.autoray)

A

abs (in module autoray.lazy) abs_ (in module autoray.lazy.core) add (in module autoray.lazy) (in module autoray.lazy.core) aesara_shape() (in module autoray.autoray) allclose() (in module autoray.autoray) angle (in module autoray.lazy) (in module autoray.lazy.core) arccos (in module autoray.lazy) (in module autoray.lazy.core) arccosh (in module autoray.lazy) (in module autoray.lazy.core) arcsin (in module autoray.lazy) (in module autoray.lazy.core) arcsinh (in module autoray.lazy) (in module autoray.lazy.core) arctan (in module autoray.lazy) (in module autoray.lazy.core) arctanh (in module autoray.lazy) (in module autoray.lazy.core) args (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) argsort() (in module autoray.lazy) (in module autoray.lazy.core) array() (in module autoray.lazy) (in module autoray.lazy.core) as_string() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) ascend (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute)

ascend() (in module autoray.lazy) (in module autoray.lazy.core) astype() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) (in module autoray) (in module autoray.autoray) AutoCompiled (class in autoray.compiler) autojit() (in module autoray) (in module autoray.compiler) AutoNamespace (class in autoray.autoray) autoray module autoray._version module autoray.autoray module autoray.compiler module autoray.experimental module autoray.experimental.complexity_tracing module autoray.lazy module autoray.lazy.core module autoray.lazy.draw module autoray.lazy.linalg module

B

backend (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) backend_like() (in module autoray) (in module autoray.autoray) binary_allow_1d_rhs_wrap() (in module autoray.autoray)

binary_dispatcher() (in module autoray.autoray) binomial() (autoray.autoray.JaxDefaultRNG method) broadcast_to() (in module autoray.lazy) (in module autoray.lazy.core) builtins_get_dtype_name() (in module autoray.autoray)

C

choice() (autoray.autoray.JaxDefaultRNG method) cholesky() (in module autoray.lazy.linalg) cholesky_lower() (in module autoray.autoray) choose_backend() (in module autoray.autoray) clip() (in module autoray.lazy) (in module autoray.lazy.core) closest_prime() (in module autoray.experimental.complexity_tracing) COLORING_SEED (in module autoray.lazy.draw) CompileJax (class in autoray.compiler) CompilePytensor (class in autoray.compiler) CompilePython (class in autoray.compiler) CompileTensorFlow (class in autoray.compiler) CompileTorch (class in autoray.compiler) CompileTorch2 (class in autoray.compiler) complex() (in module autoray.lazy) complex_() (in module autoray.lazy.core) complex_add_re_im() (in module autoray.autoray) compose() (in module autoray) (in module autoray.autoray) Composed (class in autoray.autoray) compute() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) (in module autoray.lazy) (in module autoray.lazy.core) compute_constants (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) compute_constants() (in module autoray.lazy.core) compute_cost() (in module autoray.experimental.complexity_tracing) compute_cost_scalings() (in module autoray.experimental.complexity_tracing) concatenate() (in module autoray.lazy) (in module autoray.lazy.core)

conj (in module autoray.lazy) (in module autoray.lazy.core) conj() (in module autoray) (in module autoray.autoray) cos (in module autoray.lazy) (in module autoray.lazy.core) cosh (in module autoray.lazy) (in module autoray.lazy.core) cost_eigh() (in module autoray.experimental.complexity_tracing) cost_einsum() (in module autoray.experimental.complexity_tracing) cost_linear() (in module autoray.experimental.complexity_tracing) cost_matmul() (in module autoray.experimental.complexity_tracing) cost_node() (in module autoray.experimental.complexity_tracing) cost_nothing() (in module autoray.experimental.complexity_tracing) cost_qr() (in module autoray.experimental.complexity_tracing) cost_scaling_einsum (in module autoray.experimental.complexity_tracing) cost_scaling_linalg() (in module autoray.experimental.complexity_tracing) cost_scaling_matmul (in module autoray.experimental.complexity_tracing) cost_scaling_tensordot (in module autoray.experimental.complexity_tracing) COST_SCALINGS (in module autoray.experimental.complexity_tracing) cost_svd() (in module autoray.experimental.complexity_tracing) cost_tensordot() (in module autoray.experimental.complexity_tracing) COSTS (in module autoray.experimental.complexity_tracing) count_around() (in module autoray.lazy.draw) ctf_array() (in module autoray.autoray) ctf_count_nonzero() (in module autoray.autoray) ctf_get_dtype_name() (in module autoray.autoray) ctf_to_numpy() (in module autoray.autoray) cupy_to_numpy() (in module autoray.autoray) currently_sharing() (in module autoray.lazy.core)

D

dag() (in module autoray) (in module autoray.autoray) dask_eye_wrapper() (in module autoray.autoray) dask_to_numpy() (in module autoray.autoray) default_dispatcher() (in module autoray.autoray) default_to_neutral_style() (in module autoray.lazy.draw) deps (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) depth (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) descend (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute)

descend() (in module autoray.lazy) (in module autoray.lazy.core) diag() (in module autoray.lazy) (in module autoray.lazy.core) diagonal() (in module autoray.lazy) (in module autoray.lazy.core) do() (in module autoray) (in module autoray.autoray) dtype (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) dtype_complex_equiv() (in module autoray.lazy.core) dtype_real_equiv() (in module autoray.lazy.core)

E

eig() (in module autoray.lazy.linalg) eigh() (in module autoray.lazy.linalg) einsum() (in module autoray.lazy) (in module autoray.lazy.core) einsum_dispatcher() (in module autoray.autoray) empty() (in module autoray.autoray) (in module autoray.lazy) (in module autoray.lazy.core) ensure_lazy() (in module autoray.lazy.core)

eq (in module autoray.lazy.core) equal (in module autoray.lazy) (in module autoray.lazy.core) exp (in module autoray.lazy) (in module autoray.lazy.core) expand_dims() (in module autoray.lazy) (in module autoray.lazy.core) exponential() (autoray.autoray.JaxDefaultRNG method) eye() (in module autoray.lazy) (in module autoray.lazy.core)

F

find_broadcast_shape() (in module autoray.lazy.core) find_common_backend() (in module autoray.lazy.core) find_common_dtype() (in module autoray.lazy.core) find_full_reshape() (in module autoray.lazy.core) find_lazy() (in module autoray.lazy.core) flip() (in module autoray.lazy) (in module autoray.lazy.core) floordivide (in module autoray.lazy) (in module autoray.lazy.core) fn (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property)

fn_name (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) frequencies() (in module autoray.experimental.complexity_tracing) from_data() (autoray.lazy.core.LazyArray class method) (autoray.lazy.LazyArray class method) from_shape() (autoray.lazy.core.LazyArray class method) (autoray.lazy.LazyArray class method) Function (class in autoray.lazy) (class in autoray.lazy.core)

G

ge (in module autoray.lazy.core) get_backend() (in module autoray) (in module autoray.autoray) get_common_dtype() (in module autoray) (in module autoray.autoray) get_data (in module autoray.lazy.core) get_default_colors_dict() (in module autoray.lazy.draw) get_depth (in module autoray.lazy.core) get_dtype (in module autoray.lazy.core) get_dtype_name() (in module autoray) (in module autoray.autoray) get_function() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) get_lib_fn() (in module autoray) (in module autoray.autoray)

get_namespace() (in module autoray) (in module autoray.autoray) get_nice_pos() (in module autoray.lazy.draw) get_sharing_cache() (in module autoray.lazy.core) get_sliced_size() (in module autoray.lazy.core) get_source (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) get_source() (in module autoray.lazy) (in module autoray.lazy.core) getitem() (in module autoray.lazy.core) getitem_hasher() (in module autoray.lazy.core) gt (in module autoray.lazy.core) gumbel() (autoray.autoray.JaxDefaultRNG method)

H

H (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) HAS_FA2 (in module autoray.lazy.draw) HAS_PYGRAPHVIZ (in module autoray.lazy.draw) hash_args_kwargs() (in module autoray.lazy.core) hash_to_color() (in module autoray.lazy.draw) hash_to_nvalues() (in module autoray.lazy.draw) history_fn_frequencies() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) history_max_size() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method)

history_num_nodes() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) history_peak_size() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) history_size_footprint() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) history_stats() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) history_total_size() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method)

I

identity() (in module autoray.autoray) (in module autoray.lazy) (in module autoray.lazy.core) imag (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) (in module autoray.lazy) (in module autoray.lazy.core) imag() (in module autoray) (in module autoray.autoray) import_lib_fn() (in module autoray.autoray) infer_backend() (in module autoray) (in module autoray.autoray)

infer_backend_multi() (in module autoray) (in module autoray.autoray) InjectDtypeDevice (class in autoray.autoray) integers() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) inv() (in module autoray.lazy.linalg) is_array() (in module autoray) (in module autoray.autoray) IS_CONTAINER_CACHE (in module autoray.autoray) is_lazy_array() (in module autoray.lazy.core) is_leaf_placeholder() (in module autoray.autoray) is_not_container() (in module autoray.autoray) is_prime() (in module autoray.experimental.complexity_tracing)

J

jax_default_rng() (in module autoray.autoray) jax_random_get_key() (in module autoray.autoray) jax_random_normal() (in module autoray.autoray) jax_random_seed() (in module autoray.autoray)

jax_random_uniform() (in module autoray.autoray) jax_to_numpy() (in module autoray.autoray) JaxDefaultRNG (class in autoray.autoray) join_array_dispatcher() (in module autoray.autoray)

K

kron() (in module autoray.lazy) (in module autoray.lazy.core)

kwargs (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property)

L

layout_pygraphviz() (in module autoray.lazy.draw) lazy_astype() (in module autoray.lazy.core) lazy_cache() (in module autoray.lazy.core) lazy_get_dtype_name() (in module autoray.lazy.core) LazyArray (class in autoray.lazy) (class in autoray.lazy.core) le (in module autoray.lazy.core) Leaf (class in autoray.autoray)

LEAF (in module autoray.autoray) log (in module autoray.lazy) (in module autoray.lazy.core) log10 (in module autoray.lazy) (in module autoray.lazy.core) log2 (in module autoray.lazy) (in module autoray.lazy.core) lt (in module autoray.lazy.core)

M

make_binary_func() (in module autoray.lazy.core) make_function() (autoray.autoray.Composed method) make_reduction_func() (in module autoray.lazy.core) make_translator() (in module autoray.autoray) make_unary_func() (in module autoray.lazy.core) mars_to_numpy() (in module autoray.autoray) massage_pos() (in module autoray.lazy.draw) materialize_dict() (in module autoray.lazy.core) materialize_identity() (in module autoray.lazy.core) materialize_larray() (in module autoray.lazy.core) materialize_list() (in module autoray.lazy.core) materialize_tuple() (in module autoray.lazy.core) matmul() (in module autoray.lazy) (in module autoray.lazy.core) max (in module autoray.lazy) max_ (in module autoray.lazy.core) maybe_id() (in module autoray.lazy.core) maybe_materialize() (in module autoray.lazy.core)

min (in module autoray.lazy) min_ (in module autoray.lazy.core) mod (in module autoray.lazy.core) module autoray autoray._version autoray.autoray autoray.compiler autoray.experimental autoray.experimental.complexity_tracing autoray.lazy autoray.lazy.core autoray.lazy.draw autoray.lazy.linalg multi_class_priorities (in module autoray.autoray) multiply (in module autoray.lazy) (in module autoray.lazy.core) mxnet_to_numpy() (in module autoray.autoray)

N

ndim (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) ndim() (in module autoray) (in module autoray.autoray) ne (in module autoray.lazy.core) NEUTRAL_STYLE (in module autoray.lazy.draw)

norm() (in module autoray.lazy.linalg) normal() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) nothing() (in module autoray.autoray) numpy (in module autoray) (in module autoray.autoray) numpy_to_numpy() (in module autoray.autoray)

O

ones() (in module autoray.lazy) (in module autoray.lazy.core)

P

paddle_get_dtype_name() (in module autoray.autoray) paddle_imag() (in module autoray.autoray) paddle_indices() (in module autoray.autoray) paddle_pad() (in module autoray.autoray) paddle_ravel() (in module autoray.autoray) paddle_real() (in module autoray.autoray) paddle_shape() (in module autoray.autoray) paddle_split_wrap() (in module autoray.autoray) paddle_to_numpy() (in module autoray.autoray) paddle_transpose() (in module autoray.autoray) paddle_wrap_reduction() (in module autoray.autoray) permutation() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) permute_dims (in module autoray.lazy) (in module autoray.lazy.core) Placeholder (class in autoray.lazy.core) PLACEHOLDER (in module autoray.lazy.core) plot (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) plot_circuit (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) plot_circuit() (in module autoray.lazy.draw) plot_graph (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) plot_graph() (in module autoray.lazy.draw) plot_history_functions (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute)

plot_history_functions() (in module autoray.lazy.draw) plot_history_functions_image (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) plot_history_functions_lines (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) plot_history_functions_scatter (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) plot_history_size_footprint (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) plot_history_size_footprint() (in module autoray.lazy.draw) plot_history_stats (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) plot_history_stats() (in module autoray.lazy.draw) plot_history_stats_counts (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) plot_history_stats_sizein (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) poisson() (autoray.autoray.JaxDefaultRNG method) pow_ (in module autoray.lazy.core) prime_factors() (in module autoray.experimental.complexity_tracing) print_source() (autoray.lazy.core.Function method) (autoray.lazy.Function method) prod (in module autoray.lazy) (in module autoray.lazy.core) pytensor_shape() (in module autoray.autoray) pytensor_wrap_qr_with_shapes() (in module autoray.autoray) pytensor_wrap_svd_with_shapes() (in module autoray.autoray)

Q

qr() (in module autoray.lazy.linalg)

qr_allow_fat() (in module autoray.autoray)

R

random() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) real (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) (in module autoray.lazy) (in module autoray.lazy.core) real() (in module autoray) (in module autoray.autoray) register() (autoray.autoray.Composed method) register_backend() (in module autoray) (in module autoray.autoray)

register_creation_routine() (in module autoray.autoray) register_dispatch() (in module autoray.autoray) register_function() (in module autoray) (in module autoray.autoray) reshape() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) (in module autoray) (in module autoray.autoray) (in module autoray.lazy) (in module autoray.lazy.core) rotate() (in module autoray.lazy.draw) rotated_house_shape() (in module autoray.lazy.draw)

S

scale_random_normal_manually() (in module autoray.autoray) scale_random_uniform_manually() (in module autoray.autoray) set_backend() (in module autoray) (in module autoray.autoray) set_coloring_seed() (in module autoray.lazy.draw) setup() (autoray.compiler.CompileJax method) (autoray.compiler.CompilePytensor method) (autoray.compiler.CompilePython method) (autoray.compiler.CompileTensorFlow method) (autoray.compiler.CompileTorch method) shape (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) shape() (in module autoray) (in module autoray.autoray) shared_intermediates() (in module autoray.lazy) (in module autoray.lazy.core) show() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) sign (in module autoray.lazy) (in module autoray.lazy.core) sin (in module autoray.lazy) (in module autoray.lazy.core) sinh (in module autoray.lazy) (in module autoray.lazy.core) size (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) size() (in module autoray) (in module autoray.autoray) solve() (in module autoray.lazy.linalg) sort() (in module autoray.lazy) (in module autoray.lazy.core) span() (in module autoray.lazy.draw)

sparse_array() (in module autoray.autoray) sparse_conj() (in module autoray.autoray) sparse_count_nonzero() (in module autoray.autoray) sparse_imag() (in module autoray.autoray) sparse_prod() (in module autoray.autoray) sparse_random_normal() (in module autoray.autoray) sparse_random_uniform() (in module autoray.autoray) sparse_real() (in module autoray.autoray) sparse_reshape() (in module autoray.autoray) sparse_sum() (in module autoray.autoray) sparse_to_numpy() (in module autoray.autoray) sparse_transpose() (in module autoray.autoray) split() (in module autoray.lazy) (in module autoray.lazy.core) sqrt (in module autoray.lazy) (in module autoray.lazy.core) stack() (in module autoray.lazy) (in module autoray.lazy.core) stringify() (in module autoray.lazy.core) stringify_dict() (in module autoray.lazy.core) stringify_identity() (in module autoray.lazy.core) stringify_larray() (in module autoray.lazy.core) stringify_list() (in module autoray.lazy.core) stringify_tuple() (in module autoray.lazy.core) sub (in module autoray.lazy.core) sum (in module autoray.lazy) sum_ (in module autoray.lazy.core) svd() (in module autoray.lazy.linalg) svd_manual_full_matrices_kwarg() (in module autoray.autoray) svd_not_full_matrices_wrapper() (in module autoray.autoray) svd_sUV_to_UsVH_wrapper() (in module autoray.autoray) svd_UsV_to_UsVH_wrapper() (in module autoray.autoray)

T

T (autoray.lazy.core.LazyArray property) (autoray.lazy.LazyArray property) take() (in module autoray.lazy) (in module autoray.lazy.core) take_along_axis() (in module autoray.lazy) (in module autoray.lazy.core) tan (in module autoray.lazy) (in module autoray.lazy.core) tanh (in module autoray.lazy) (in module autoray.lazy.core) tensordot() (in module autoray.lazy) (in module autoray.lazy.core) tensorflow_indices() (in module autoray.autoray) tensorflow_pad_wrap() (in module autoray.autoray) tensorflow_to_numpy() (in module autoray.autoray) to() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) to_backend_dtype() (in module autoray) (in module autoray.autoray) to_numpy() (in module autoray) (in module autoray.autoray) to_nx_digraph() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) to_queue() (in module autoray.lazy.core) torch_astype() (in module autoray.autoray) torch_copy() (in module autoray.autoray) torch_count_nonzero() (in module autoray.autoray) torch_default_rng() (in module autoray.autoray) torch_eye_wrap() (in module autoray.autoray) torch_flip_wrap() (in module autoray.autoray) torch_get_dtype_name() (in module autoray.autoray) torch_imag() (in module autoray.autoray) torch_indices() (in module autoray.autoray) torch_linalg_eigh() (in module autoray.autoray) torch_linalg_eigvalsh() (in module autoray.autoray) torch_linalg_solve_wrap() (in module autoray.autoray) torch_pad() (in module autoray.autoray) torch_real() (in module autoray.autoray) torch_shape() (in module autoray.autoray) torch_size() (in module autoray.autoray) torch_sort_wrap() (in module autoray.autoray) torch_split_wrap() (in module autoray.autoray)

torch_tensordot_wrap() (in module autoray.autoray) torch_to_numpy() (in module autoray.autoray) torch_transpose() (in module autoray.autoray) torch_zeros_ones_wrap() (in module autoray.autoray) TorchDefaultRNG (class in autoray.autoray) trace() (in module autoray.lazy) (in module autoray.lazy.core) translate_wrapper() (in module autoray.autoray) transpose() (in module autoray) (in module autoray.autoray) (in module autoray.lazy) (in module autoray.lazy.core) TREE_APPLIER_CACHE (in module autoray.autoray) tree_apply() (in module autoray) (in module autoray.autoray) tree_apply_dict() (in module autoray.autoray) tree_apply_list() (in module autoray.autoray) TREE_APPLY_REGISTRY (in module autoray.autoray) tree_apply_tuple() (in module autoray.autoray) tree_flatten() (in module autoray) (in module autoray.autoray) tree_iter() (in module autoray) (in module autoray.autoray) TREE_ITER_CACHE (in module autoray.autoray) tree_iter_dict() (in module autoray.autoray) tree_iter_list() (in module autoray.autoray) TREE_ITER_REGISTRY (in module autoray.autoray) tree_iter_tuple() (in module autoray.autoray) tree_map() (in module autoray) (in module autoray.autoray) tree_map_dict() (in module autoray.autoray) tree_map_list() (in module autoray.autoray) TREE_MAP_REGISTRY (in module autoray.autoray) tree_map_tuple() (in module autoray.autoray) TREE_MAPPER_CACHE (in module autoray.autoray) tree_register_container() (in module autoray.autoray) tree_unflatten() (in module autoray) (in module autoray.autoray) tril_to_band_part() (in module autoray.autoray) triu_to_band_part() (in module autoray.autoray) truedivide (in module autoray.lazy) (in module autoray.lazy.core) TYPE_CHECKING (in module autoray._version)

U

uniform() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.TorchDefaultRNG method)

V

Variable() (in module autoray.lazy) (in module autoray.lazy.core)

version (in module autoray._version) VERSION_TUPLE (in module autoray._version) version_tuple (in module autoray._version)

W

where() (in module autoray.lazy) (in module autoray.lazy.core)

with_dtype_wrapper() (in module autoray.autoray) wrap_args_kwargs_from_raw() (in module autoray.autoray)

Z

zeros() (in module autoray.lazy) (in module autoray.lazy.core)