Index _ | A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | Z _ __abs__() (autoray.lazy.core.LazyArray method) __add__() (autoray.lazy.core.LazyArray method) __array_namespace__() (autoray.lazy.core.LazyArray method) __array_ufunc__ (autoray.lazy.core.LazyArray attribute) __call__() (autoray.autoray.Composed method) (autoray.autoray.ConstantInferrer method) (autoray.autoray.DoFunc 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) __commit_id__ (in module autoray._version) __eq__() (autoray.lazy.core.LazyArray method) __floordiv__() (autoray.lazy.core.LazyArray method) __ge__() (autoray.lazy.core.LazyArray method) __getattr__() (autoray.autoray.AutoNamespace method) __getitem__() (autoray.lazy.core.LazyArray method) __getstate__() (autoray.lazy.core.Function method) __gt__() (autoray.lazy.core.LazyArray method) __hash__() (autoray.lazy.core.LazyArray method) __iter__() (autoray.lazy.core.LazyArray method) __le__() (autoray.lazy.core.LazyArray method) __len__() (autoray.lazy.core.LazyArray method) __lt__() (autoray.lazy.core.LazyArray method) __matmul__() (autoray.lazy.core.LazyArray method) __mod__() (autoray.lazy.core.LazyArray method) __mul__() (autoray.lazy.core.LazyArray method) __ne__() (autoray.lazy.core.LazyArray method) __neg__() (autoray.lazy.core.LazyArray method) __pow__() (autoray.lazy.core.LazyArray method) __radd__() (autoray.lazy.core.LazyArray method) __repr__() (autoray.autoray.AutoNamespace method) (autoray.autoray.Composed method) (autoray.autoray.DoFunc method) (autoray.autoray.InjectDtypeDevice method) (autoray.autoray.Leaf method) (autoray.lazy.core.Function method) (autoray.lazy.core.LazyArray method) (autoray.lazy.core.Placeholder method) __rfloordiv__() (autoray.lazy.core.LazyArray method) __rmatmul__() (autoray.lazy.core.LazyArray method) __rmul__() (autoray.lazy.core.LazyArray method) __rpow__() (autoray.lazy.core.LazyArray method) __rsub__() (autoray.lazy.core.LazyArray method) __rtruediv__() (autoray.lazy.core.LazyArray method) __setstate__() (autoray.lazy.core.Function method) __slots__ (autoray.autoray.ConstantInferrer attribute) (autoray.autoray.DoFunc attribute) (autoray.autoray.InjectDtypeDevice attribute) (autoray.autoray.Leaf attribute) (autoray.lazy.core.Function attribute) (autoray.lazy.core.LazyArray attribute) (autoray.lazy.core.Placeholder attribute) __sub__() (autoray.lazy.core.LazyArray method) __truediv__() (autoray.lazy.core.LazyArray method) __version__ (in module autoray._version) __version_tuple__ (in module autoray._version) _add_sharing_cache() (in module autoray.lazy.core) _always_false() (in module autoray.autoray) _always_true() (in module autoray.autoray) _args (autoray.lazy.core.LazyArray attribute) _backend (autoray.compiler.AutoCompiled attribute) (autoray.lazy.core.LazyArray attribute) _BACKEND_ALIASES (in module autoray.autoray) _backend_device_dtype_dispatchers (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) _BUILTIN_DTYPE_NAMES (in module autoray.autoray) _choose_backend() (in module autoray.autoray) _choose_namespace() (in module autoray.autoray) _code (autoray.lazy.core.Function attribute) _compiled_fns (autoray.compiler.AutoCompiled attribute) _compiler_lookup (in module autoray.compiler) _COMPLEX_DTYPES (in module autoray.autoray) _COMPLEX_TO_REAL_DTYPE (in module autoray.autoray) _COMPOSED_FUNCTION_GENERATORS (in module autoray.autoray) _CREATION_INJECT (in module autoray.autoray) _CREATION_ROUTINES (in module autoray.autoray) _cupy_parse_device() (in module autoray.autoray) _CUSTOM_BACKENDS (in module autoray.autoray) _CUSTOM_WRAPPERS (in module autoray.autoray) _data (autoray.lazy.core.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.InjectDtypeDevice attribute) _DEVICE_MATCHER (in module autoray.autoray) _DISPATCHERS (in module autoray.autoray) _DOUBLE_DTYPES (in module autoray.autoray) _dtype (autoray.autoray.InjectDtypeDevice attribute) _dtype_is_inexact() (in module autoray.autoray) _DTYPE_MAP (in module autoray.autoray) _DTYPE_MATCHER (in module autoray.autoray) _dtype_to_name_cached() (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) _fold_constants (autoray.compiler.CompilePython attribute) _FUNC_ALIASES (in module autoray.autoray) _FUNCS (in module autoray.autoray) _get_fn() (autoray.autoray.AutoNamespace method) _get_jax_default_rng() (in module autoray.autoray) _get_mlx_shared_rng() (in module autoray.autoray) _get_parse_einsum_input() (in module autoray.lazy.core) _get_py_shape() (in module autoray.lazy.core) _get_rademacher_table() (in module autoray.autoray) _get_root_namespace() (autoray.autoray.AutoNamespace method) _get_submodule() (autoray.autoray.AutoNamespace method) _global_backend (in module autoray.autoray) _global_backends_threadaware (in module autoray.autoray) _GRAD_BACKENDS (in module autoray.grad) _handle_size_to_shape() (in module autoray.autoray) _importing_thrid (in module autoray.autoray) _in_names (autoray.lazy.core.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) _invalidate_backend_inference_caches() (in module autoray.autoray) _IS_SCALAR_CACHE (in module autoray.autoray) _is_scalar_ndim() (in module autoray.autoray) _JAX_DEFAULT_RNG (in module autoray.autoray) _jax_parse_device() (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) _make_default_function() (autoray.autoray.Composed method) _make_device_dtype_dispatch() (in module autoray.autoray) _materialize() (autoray.lazy.core.LazyArray method) _materialize_dispatch (in module autoray.lazy.core) _MODULE_ALIASES (in module autoray.autoray) _name (autoray.autoray.Composed attribute) _NAME_SPACE_SUBMODULES (in module autoray.autoray) _NAMESPACE_ATTRS (in module autoray.autoray) _NAMESPACE_CACHE (in module autoray.autoray) _namespace_key_part() (in module autoray.autoray) _NO_GRAD_BACKENDS (in module autoray.grad) _out_names (autoray.lazy.core.Function attribute) _output_ref_tree (autoray.compiler.CompilePytensor attribute) _paddle_dtype_name_conversion (in module autoray.autoray) _parse_compound_backend_spec() (in module autoray.autoray) _platform_name (autoray.compiler.CompileJax attribute) _rademacher_cupy() (in module autoray.autoray) _rademacher_jax() (in module autoray.autoray) _rademacher_numpy() (in module autoray.autoray) _RADEMACHER_ROOTS (in module autoray.autoray) _RADEMACHER_SAMPLERS (in module autoray.autoray) _RADEMACHER_SHARED (in module autoray.autoray) _RADEMACHER_SIGNS (in module autoray.autoray) _rademacher_torch() (in module autoray.autoray) _RANDOM_DISTS (in module autoray.autoray) _register_creation_inject() (in module autoray.autoray) _remove_sharing_cache() (in module autoray.lazy.core) _reset_namespace() (in module autoray.autoray) _reset_namespaces() (in module autoray.autoray) _reshape_tuple() (in module autoray.lazy.core) _resolve_dtype() (autoray.autoray.MlxDefaultRNG method) _sample_rademacher() (in module autoray.autoray) _sample_random_array() (in module autoray.autoray) _set_default_device() (autoray.autoray.TorchDefaultRNG method) _set_default_dtype() (autoray.autoray.TorchDefaultRNG method) _shape (autoray.lazy.core.LazyArray attribute) _share_intermediates (autoray.compiler.CompilePython attribute) _SHARING_STACK (in module autoray.lazy.core) _source (autoray.lazy.core.Function attribute) _split_key() (autoray.autoray.MlxDefaultRNG method) (autoray.autoray.MlxSharedRNG method) _stringify_dispatch (in module autoray.lazy.core) _submodule (autoray.autoray.AutoNamespace attribute) _SUBMODULE_ALIASES (in module autoray.autoray) _supply_backend (autoray.autoray.Composed attribute) _supply_namespace (autoray.autoray.Composed attribute) _tensorflow_translate_device() (in module autoray.autoray) _to_backend_dtype_from_str_cached() (in module autoray.autoray) _torch (autoray.autoray.TorchDefaultRNG attribute) _torch_reduce_translation (in module autoray.autoray) _warn_jax_generated_seed() (in module autoray.autoray) A abs_ (in module autoray.lazy.core) add (in module autoray.lazy.core) aesara_shape() (in module autoray.autoray) all_ (in module autoray.lazy.core) allclose() (in module autoray.autoray) (in module autoray.lazy.core) angle (in module autoray.lazy.core) any_ (in module autoray.lazy.core) arccos (in module autoray.lazy.core) arccosh (in module autoray.lazy.core) arcsin (in module autoray.lazy.core) arcsinh (in module autoray.lazy.core) arctan (in module autoray.lazy.core) arctanh (in module autoray.lazy.core) argmax (in module autoray.lazy.core) argmin (in module autoray.lazy.core) args (autoray.lazy.core.LazyArray property) argsort() (in module autoray.lazy.core) array() (in module autoray.lazy.core) as_string() (autoray.lazy.core.LazyArray method) as_string_pretty() (autoray.lazy.core.LazyArray method) ascend (autoray.lazy.core.LazyArray attribute) ascend() (in module autoray.lazy.core) astype() (autoray.lazy.core.LazyArray method) (in module autoray.autoray) AutoCompiled (class in autoray.compiler) autograd_stop_gradient() (in module autoray.grad) autograd_take() (in module autoray.autoray) autojit() (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.grad module autoray.lazy module autoray.lazy.core module autoray.lazy.draw module autoray.lazy.linalg module B backend (autoray.autoray.ConstantInferrer attribute) (autoray.lazy.core.LazyArray property) backend_like() (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.core) builtins_get_dtype_name() (in module autoray.autoray) C choice() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) cholesky() (in module autoray.lazy.linalg) cholesky_lower() (in module autoray.autoray) cholesky_manual_upper() (in module autoray.autoray) choose_backend() (in module autoray.autoray) clip() (in module autoray.lazy.core) closest_prime() (in module autoray.experimental.complexity_tracing) COLORING_SEED (in module autoray.lazy.draw) commit_id (in module autoray._version) 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.core) complex_add_re_im() (in module autoray.autoray) compose() (in module autoray.autoray) Composed (class in autoray.autoray) compute() (autoray.lazy.core.LazyArray method) (in module autoray.lazy.core) compute_constants (autoray.lazy.core.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.core) conj (in module autoray.lazy.core) conj() (in module autoray.autoray) ConstantInferrer (class in autoray.autoray) cos (in module autoray.lazy.core) cosh (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) count_nonzero (in module autoray.lazy.core) 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) cumsum() (in module autoray.lazy.core) cupy_from_numpy() (in module autoray.autoray) cupy_to_device() (in module autoray.autoray) cupy_to_numpy() (in module autoray.autoray) currently_sharing() (in module autoray.lazy.core) D dag() (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) depth (autoray.lazy.core.LazyArray property) descend (autoray.lazy.core.LazyArray attribute) descend() (in module autoray.lazy.core) diag() (in module autoray.lazy.core) diagonal() (in module autoray.lazy.core) do() (in module autoray.autoray) do_nothing() (in module autoray.grad) DoFunc (class in autoray.autoray) dtype (autoray.lazy.core.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.core) einsum_dispatcher() (in module autoray.autoray) empty() (in module autoray.autoray) (in module autoray.lazy.core) ensure_lazy() (in module autoray.lazy.core) eq (in module autoray.lazy.core) equal (in module autoray.lazy.core) exp (in module autoray.lazy.core) expand_dims() (in module autoray.lazy.core) exponential() (autoray.autoray.JaxDefaultRNG method) eye() (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.core) floordivide (in module autoray.lazy.core) fn (autoray.autoray.DoFunc attribute) (autoray.lazy.core.LazyArray property) fn_name (autoray.lazy.core.LazyArray property) frequencies() (in module autoray.experimental.complexity_tracing) from_data() (autoray.lazy.core.LazyArray class method) from_key() (autoray.autoray.JaxDefaultRNG class method) from_numpy() (in module autoray.autoray) from_shape() (autoray.lazy.core.LazyArray class method) Function (class in autoray.lazy.core) G ge (in module autoray.lazy.core) get_backend() (in module autoray.autoray) get_common_dtype() (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.autoray) get_function() (autoray.lazy.core.LazyArray method) get_jax() (in module autoray.autoray) get_lib_fn() (in module autoray.autoray) get_mlx() (in module autoray.autoray) get_namespace() (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) get_source() (in module autoray.lazy.core) get_tensorflow() (in module autoray.autoray) get_torch() (in module autoray.autoray) getitem() (in module autoray.lazy.core) getitem_hasher() (in module autoray.lazy.core) gt (in module autoray.lazy.core) gumbel() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.MlxDefaultRNG method) H H (autoray.lazy.core.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) history_max_size() (autoray.lazy.core.LazyArray method) history_num_nodes() (autoray.lazy.core.LazyArray method) history_peak_size() (autoray.lazy.core.LazyArray method) history_size_footprint() (autoray.lazy.core.LazyArray method) history_stats() (autoray.lazy.core.LazyArray method) history_total_size() (autoray.lazy.core.LazyArray method) I identity() (in module autoray.autoray) (in module autoray.lazy.core) imag (autoray.lazy.core.LazyArray property) (in module autoray.lazy.core) imag() (in module autoray.autoray) import_lib_fn() (in module autoray.autoray) infer_backend() (in module autoray.autoray) infer_backend_device_dtype() (in module autoray.autoray) infer_backend_multi() (in module autoray.autoray) InjectDtypeDevice (class in autoray.autoray) integers() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.MlxDefaultRNG method) (autoray.autoray.TensorflowDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) inv() (in module autoray.lazy.linalg) is_array() (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) is_scalar() (in module autoray.autoray) J jax (autoray.autoray.JaxDefaultRNG attribute) jax_default_rng() (in module autoray.autoray) jax_from_numpy() (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_stop_gradient() (in module autoray.grad) jax_to_device() (in module autoray.autoray) JaxDefaultRNG (class in autoray.autoray) join_array_dispatcher() (in module autoray.autoray) K key (autoray.autoray.JaxDefaultRNG attribute) (autoray.autoray.MlxDefaultRNG attribute) kron() (in module autoray.lazy.core) kwargs (autoray.lazy.core.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_from_numpy() (in module autoray.lazy.core) lazy_get_dtype_name() (in module autoray.lazy.core) lazy_to_device() (in module autoray.lazy.core) lazy_to_numpy() (in module autoray.lazy.core) LazyArray (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.core) log10 (in module autoray.lazy.core) log2 (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) 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.core) max_ (in module autoray.lazy.core) maybe_id() (in module autoray.lazy.core) maybe_materialize() (in module autoray.lazy.core) mean (in module autoray.lazy.core) min_ (in module autoray.lazy.core) mlx_array_asarray_wrap() (in module autoray.autoray) mlx_count_nonzero() (in module autoray.autoray) mlx_default_rng() (in module autoray.autoray) mlx_eye_wrap() (in module autoray.autoray) mlx_random_normal() (in module autoray.autoray) mlx_random_seed() (in module autoray.autoray) mlx_random_uniform() (in module autoray.autoray) mlx_ravel() (in module autoray.autoray) mlx_to_device() (in module autoray.autoray) mlx_zeros_ones_wrap() (in module autoray.autoray) MlxDefaultRNG (class in autoray.autoray) MlxSharedRNG (class in autoray.autoray) mod (in module autoray.lazy.core) module autoray autoray._version autoray.autoray autoray.compiler autoray.experimental autoray.experimental.complexity_tracing autoray.grad autoray.lazy autoray.lazy.core autoray.lazy.draw autoray.lazy.linalg moveaxis() (in module autoray.lazy.core) multi_class_priorities (in module autoray.autoray) multiply (in module autoray.lazy.core) mx (autoray.autoray.MlxDefaultRNG attribute) (autoray.autoray.MlxSharedRNG attribute) mxnet_to_numpy() (in module autoray.autoray) N ndim (autoray.lazy.core.LazyArray property) ndim() (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.MlxDefaultRNG method) (autoray.autoray.TensorflowDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) nothing() (in module autoray.autoray) numpy (in module autoray.autoray) O ones() (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_stop_gradient() (in module autoray.grad) paddle_to_numpy() (in module autoray.autoray) paddle_transpose() (in module autoray.autoray) paddle_wrap_reduction() (in module autoray.autoray) parse_creation_shape() (in module autoray.lazy.core) permutation() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.MlxDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) permute_dims (in module autoray.lazy.core) Placeholder (class in autoray.lazy.core) PLACEHOLDER (in module autoray.lazy.core) plot (autoray.lazy.core.LazyArray attribute) plot_circuit (autoray.lazy.core.LazyArray attribute) plot_circuit() (in module autoray.lazy.draw) plot_graph (autoray.lazy.core.LazyArray attribute) plot_graph() (in module autoray.lazy.draw) plot_history_functions (autoray.lazy.core.LazyArray attribute) plot_history_functions() (in module autoray.lazy.draw) plot_history_functions_image (autoray.lazy.core.LazyArray attribute) plot_history_functions_lines (autoray.lazy.core.LazyArray attribute) plot_history_functions_scatter (autoray.lazy.core.LazyArray attribute) plot_history_size_footprint (autoray.lazy.core.LazyArray attribute) plot_history_size_footprint() (in module autoray.lazy.draw) plot_history_stats (autoray.lazy.core.LazyArray attribute) plot_history_stats() (in module autoray.lazy.draw) plot_history_stats_counts (autoray.lazy.core.LazyArray attribute) plot_history_stats_sizein (autoray.lazy.core.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) prod (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.MlxDefaultRNG method) (autoray.autoray.TensorflowDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) random_array() (in module autoray.autoray) random_array_dispatcher() (in module autoray.autoray) real (autoray.lazy.core.LazyArray property) (in module autoray.lazy.core) real() (in module autoray.autoray) register() (autoray.autoray.Composed method) register_backend() (in module autoray.autoray) register_backend_alias() (in module autoray.autoray) register_creation_routine() (in module autoray.autoray) register_custom_wrapper() (in module autoray.autoray) register_dispatch() (in module autoray.autoray) register_func_alias() (in module autoray.autoray) register_function() (in module autoray.autoray) register_module_alias() (in module autoray.autoray) register_submodule_alias() (in module autoray.autoray) reshape() (autoray.lazy.core.LazyArray method) (in module autoray.autoray) (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.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) shape() (in module autoray.autoray) shared_intermediates() (in module autoray.lazy.core) show() (autoray.lazy.core.LazyArray method) sign (in module autoray.lazy.core) sin (in module autoray.lazy.core) sinh (in module autoray.lazy.core) size (autoray.lazy.core.LazyArray property) size() (in module autoray.autoray) solve() (in module autoray.lazy.linalg) sort() (in module autoray.lazy.core) span() (in module autoray.lazy.draw) sparse_array() (in module autoray.autoray) sparse_complex() (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.core) sqrt (in module autoray.lazy.core) stack() (in module autoray.lazy.core) std (in module autoray.lazy.core) stop_gradient() (in module autoray.grad) 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.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) swapaxes() (in module autoray.lazy.core) T T (autoray.lazy.core.LazyArray property) take() (in module autoray.lazy.core) take_along_axis() (in module autoray.lazy.core) tan (in module autoray.lazy.core) tanh (in module autoray.lazy.core) tensordot() (in module autoray.lazy.core) tensorflow_astype() (in module autoray.autoray) tensorflow_cholesky() (in module autoray.autoray) tensorflow_default_rng() (in module autoray.autoray) tensorflow_from_numpy() (in module autoray.autoray) tensorflow_indices() (in module autoray.autoray) tensorflow_pad_wrap() (in module autoray.autoray) tensorflow_solve_triangular() (in module autoray.autoray) tensorflow_stop_gradient() (in module autoray.grad) tensorflow_swapaxes() (in module autoray.autoray) tensorflow_to_device() (in module autoray.autoray) tensorflow_to_numpy() (in module autoray.autoray) tensorflow_wrap_norm() (in module autoray.autoray) TensorflowDefaultRNG (class in autoray.autoray) tf (autoray.autoray.TensorflowDefaultRNG attribute) to() (autoray.lazy.core.LazyArray method) (in module autoray.autoray) to_backend_dtype() (in module autoray.autoray) to_device() (in module autoray.autoray) to_numpy() (in module autoray.autoray) to_nx_digraph() (autoray.lazy.core.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_complex() (in module autoray.autoray) torch_copy() (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_from_numpy() (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_maybe_convert_dtype_from_str() (in module autoray.autoray) torch_nonzero_wrap() (in module autoray.autoray) torch_pad() (in module autoray.autoray) torch_random_normal() (in module autoray.autoray) torch_random_seed() (in module autoray.autoray) torch_random_uniform() (in module autoray.autoray) torch_real() (in module autoray.autoray) torch_scipy_linalg_solve_triangular() (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_stop_gradient() (in module autoray.grad) torch_take() (in module autoray.autoray) torch_tensordot_wrap() (in module autoray.autoray) torch_to_device() (in module autoray.autoray) torch_to_numpy() (in module autoray.autoray) torch_trace() (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.core) translate_wrapper() (in module autoray.autoray) transpose() (in module autoray.autoray) (in module autoray.lazy.core) TREE_APPLIER_CACHE (in module autoray.autoray) tree_apply() (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.autoray) tree_iter() (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.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.autoray) tril_to_band_part() (in module autoray.autoray) triu_to_band_part() (in module autoray.autoray) truedivide (in module autoray.lazy.core) U uniform() (autoray.autoray.JaxDefaultRNG method) (autoray.autoray.MlxDefaultRNG method) (autoray.autoray.TensorflowDefaultRNG method) (autoray.autoray.TorchDefaultRNG method) V var (in module autoray.lazy.core) Variable() (in module autoray.lazy.core) version (in module autoray._version) version_tuple (in module autoray._version) W where() (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.core)