LICENSE
README.md
pyproject.toml
einx/__init__.py
einx/arrayapi.py
einx/backend.py
einx/errors.py
einx/jax.py
einx/mlx.py
einx/nn.py
einx/numpy.py
einx/tensorflow.py
einx/tinygrad.py
einx/torch.py
einx.egg-info/PKG-INFO
einx.egg-info/SOURCES.txt
einx.egg-info/dependency_links.txt
einx.egg-info/requires.txt
einx.egg-info/top_level.txt
einx/_src/adapter/__init__.py
einx/_src/adapter/_util.py
einx/_src/adapter/classical_from_classical.py
einx/_src/adapter/classical_from_einsum.py
einx/_src/adapter/decomposednamedtensor_from_classical.py
einx/_src/adapter/decomposednamedtensor_from_einsum.py
einx/_src/adapter/decomposednamedtensor_from_vmap.py
einx/_src/adapter/einx_from_namedtensor.py
einx/_src/adapter/elementary_from_classical.py
einx/_src/adapter/namedtensor_calltensorfactory.py
einx/_src/adapter/namedtensor_from_decomposednamedtensor.py
einx/_src/adapter/ops.py
einx/_src/adapter/arrayapi/classical_from_arrayapi.py
einx/_src/adapter/arrayapi/einsum_from_arrayapi.py
einx/_src/adapter/arrayapi/namespacestack.py
einx/_src/adapter/arrayapi/tensortype_from_arrayapi.py
einx/_src/adapter/functorchdim/namedtensor_from_functorchdim.py
einx/_src/adapter/jax/classical_from_jax.py
einx/_src/adapter/jax/einsum_from_jax.py
einx/_src/adapter/jax/vmap_from_jax.py
einx/_src/adapter/mlx/classical_from_mlx.py
einx/_src/adapter/mlx/einsum_from_mlx.py
einx/_src/adapter/mlx/vmap_from_mlx.py
einx/_src/adapter/numpy/classical_from_numpy.py
einx/_src/adapter/numpy/einsum_from_numpy.py
einx/_src/adapter/tensorflow/classical_from_tensorflow.py
einx/_src/adapter/tensorflow/einsum_from_tensorflow.py
einx/_src/adapter/tinygrad/classical_from_tinygrad.py
einx/_src/adapter/tinygrad/einsum_from_tinygrad.py
einx/_src/adapter/torch/classical_from_torch.py
einx/_src/adapter/torch/devicestack.py
einx/_src/adapter/torch/einsum_from_torch.py
einx/_src/adapter/torch/vmap_from_torch.py
einx/_src/frontend/api.py
einx/_src/frontend/backend.py
einx/_src/frontend/errors.py
einx/_src/frontend/ops.py
einx/_src/frontend/removed_ops.py
einx/_src/frontend/types.py
einx/_src/frontend/util.py
einx/_src/frontend/impl/_docs.py
einx/_src/frontend/impl/_util.py
einx/_src/frontend/impl/arrayapi.py
einx/_src/frontend/impl/functorchdim.py
einx/_src/frontend/impl/jax.py
einx/_src/frontend/impl/mlx.py
einx/_src/frontend/impl/numpy.py
einx/_src/frontend/impl/tensorflow.py
einx/_src/frontend/impl/tinygrad.py
einx/_src/frontend/impl/torch.py
einx/_src/namedtensor/__init__.py
einx/_src/namedtensor/namedtensor.py
einx/_src/namedtensor/solve.py
einx/_src/namedtensor/util.py
einx/_src/namedtensor/stage1/__init__.py
einx/_src/namedtensor/stage1/parse.py
einx/_src/namedtensor/stage1/transform.py
einx/_src/namedtensor/stage1/tree.py
einx/_src/namedtensor/stage2/__init__.py
einx/_src/namedtensor/stage2/cse.py
einx/_src/namedtensor/stage2/solve.py
einx/_src/namedtensor/stage2/transform.py
einx/_src/namedtensor/stage2/tree.py
einx/_src/namedtensor/stage3/__init__.py
einx/_src/namedtensor/stage3/solve.py
einx/_src/namedtensor/stage3/transform.py
einx/_src/namedtensor/stage3/tree.py
einx/_src/tracer/__init__.py
einx/_src/tracer/graph.py
einx/_src/tracer/visualize.py
einx/_src/tracer/compiler/__init__.py
einx/_src/tracer/compiler/run.py
einx/_src/tracer/compiler/python/__init__.py
einx/_src/tracer/compiler/python/scope.py
einx/_src/tracer/compiler/python/usage.py
einx/_src/tracer/optimizer/__init__.py
einx/_src/tracer/optimizer/_util.py
einx/_src/tracer/optimizer/classical.py
einx/_src/tracer/optimizer/graph.py
einx/_src/tracer/optimizer/optimizer.py
einx/_src/tracer/signature/__init__.py
einx/_src/tracer/signature/python.py
einx/_src/tracer/signature/classical/__init__.py
einx/_src/tracer/signature/classical/arrayapi.py
einx/_src/tracer/signature/classical/functions.py
einx/_src/tracer/signature/classical/jax.py
einx/_src/tracer/signature/classical/mlx.py
einx/_src/tracer/signature/classical/numpy.py
einx/_src/tracer/signature/classical/tensor.py
einx/_src/tracer/signature/classical/tensorflow.py
einx/_src/tracer/signature/classical/tinygrad.py
einx/_src/tracer/signature/classical/torch.py
einx/_src/tracer/signature/functorchdim/__init__.py
einx/_src/tracer/signature/functorchdim/tensor.py
einx/_src/util/functools.py
einx/_src/util/lru_cache.py
einx/_src/util/pytree.py
einx/_src/util/rwlock.py
einx/_src/util/solver.py
einx/experimental/__init__.py
einx/experimental/functorchdim.py
test/test_adapt.py
test/test_backend_resolution.py
test/test_compare_libs.py
test/test_compilation.py
test/test_experimental.py
test/test_invalid_backend.py
test/test_invalid_backend_version.py
test/test_shapes.py
test/test_tensor_conversion.py
test/test_values.py