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) _args (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) _backend (autoray.compiler.AutoCompiled attribute) (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute)

_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) _code (autoray.lazy.core.Function attribute) (autoray.lazy.Function attribute) _compiled_fns (autoray.compiler.AutoCompiled attribute) _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) _data (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) _default_fn (autoray.autoray.Composed attribute) _default_infer_from_sig() (in module autoray.autoray) _default_infer_from_sig_threadaware() (in module autoray.autoray) _device (autoray.autoray.AutoNamespace attribute) (autoray.autoray.InjectDtypeDevice attribute) _DISPATCHERS (in module autoray.autoray) _DOUBLE_DTYPES (in module autoray.autoray) _dtype (autoray.autoray.AutoNamespace attribute) (autoray.autoray.InjectDtypeDevice attribute) _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) _enable_x64 (autoray.compiler.CompileJax attribute) _find_common_backend_cached() (in module autoray.lazy.core) _find_common_dtype() (in module autoray.lazy.core) _fn (autoray.autoray.InjectDtypeDevice attribute) (autoray.compiler.AutoCompiled attribute) (autoray.compiler.CompileJax attribute) (autoray.compiler.CompilePytensor attribute) (autoray.compiler.CompilePython attribute) (autoray.compiler.CompileTensorFlow attribute) (autoray.compiler.CompileTorch attribute) (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) _fold_constants (autoray.compiler.CompilePython attribute) _FUNC_ALIASES (in module autoray.autoray) _FUNCS (in module autoray.autoray) _generator (autoray.autoray.TorchDefaultRNG attribute) _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) _in_names (autoray.lazy.core.Function attribute) (autoray.lazy.Function attribute) _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) _jit_fn (autoray.compiler.CompileJax attribute) (autoray.compiler.CompilePytensor attribute) (autoray.compiler.CompilePython attribute) (autoray.compiler.CompileTensorFlow attribute) (autoray.compiler.CompileTorch attribute) (autoray.compiler.CompileTorch2 attribute) _jit_kwargs (autoray.compiler.CompileJax attribute) (autoray.compiler.CompilePytensor attribute) (autoray.compiler.CompileTensorFlow attribute) (autoray.compiler.CompileTorch attribute) _locals (autoray.lazy.core.Function attribute) (autoray.lazy.Function attribute) _materialize() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) _materialize_dispatch (in module autoray.lazy.core) _MODULE_ALIASES (in module autoray.autoray) _name (autoray.autoray.Composed attribute) _out_names (autoray.lazy.core.Function attribute) (autoray.lazy.Function attribute) _output_ref_tree (autoray.compiler.CompilePytensor attribute) _paddle_dtype_name_conversion (in module autoray.autoray) _platform_name (autoray.compiler.CompileJax attribute) _remove_sharing_cache() (in module autoray.lazy.core) _reshape_tuple() (in module autoray.lazy.core) _shape (autoray.lazy.core.LazyArray attribute) (autoray.lazy.LazyArray attribute) _share_intermediates (autoray.compiler.CompilePython attribute) _SHARING_STACK (in module autoray.lazy.core) _source (autoray.lazy.core.Function attribute) (autoray.lazy.Function attribute) _stringify_dispatch (in module autoray.lazy.core) _submodule (autoray.autoray.AutoNamespace attribute) _SUBMODULE_ALIASES (in module autoray.autoray) _supply_backend (autoray.autoray.Composed attribute) _torch (autoray.autoray.TorchDefaultRNG attribute) _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 (autoray.autoray.JaxDefaultRNG attribute) 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

key (autoray.autoray.JaxDefaultRNG attribute) 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 (autoray.compiler.CompileTorch attribute) (autoray.compiler.CompileTorch2 attribute) 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)

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)

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)