Index

_

__abs__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __add__() (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.compiler.AutoCompiled method) (autoray.compiler.CompileJax method) (autoray.compiler.CompilePython method) (autoray.compiler.CompileTensorFlow method) (autoray.compiler.CompileTorch method) (autoray.lazy.core.Function method) (autoray.lazy.Function method) __floordiv__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __ge__() (autoray.lazy.core.LazyArray method) (autoray.lazy.LazyArray method) __getattribute__() (autoray.autoray.NumpyMimic 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) __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) __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.Composed method) (autoray.autoray.Leaf method) (autoray.autoray.NumpyMimic static 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.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) _compiler_lookup (in module autoray.compiler) _COMPLEX_DTYPES (in module autoray.autoray) _COMPOSED_FUNCTION_GENERATORS (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_parse_einsum_input() (in module autoray.lazy.core) _get_py_shape() (in module autoray.lazy.core) _global_backend (in module autoray.autoray) _global_backends_threadaware (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) _partial_functions (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)

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) 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) builtins_get_dtype_name() (in module autoray.autoray)

C

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) CompilePython (class in autoray.compiler) CompileTensorFlow (class in autoray.compiler) CompileTorch (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) do() (in module autoray) (in module autoray.autoray) 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) ensure_lazy() (in module autoray.lazy.core) exp (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_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_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)

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) 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) 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_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) 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) 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)

nothing() (in module autoray.autoray) numpy (in module autoray) (in module autoray.autoray) numpy_linalg (in module autoray.autoray) numpy_random (in module autoray.autoray) numpy_to_numpy() (in module autoray.autoray) NumpyMimic (class in autoray.autoray)

P

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) 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)

Q

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

qr_allow_fat() (in module autoray.autoray)

R

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_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.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) 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_pad_wrap() (in module autoray.autoray) tensorflow_split_wrap() (in module autoray.autoray) tensorflow_to_numpy() (in module autoray.autoray) tensorflow_where_wrap() (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_count_nonzero() (in module autoray.autoray) torch_eye_wrap() (in module autoray.autoray) torch_get_dtype_name() (in module autoray.autoray) torch_imag() (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_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) 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)

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)