LICENSE
README.md
setup.cfg
setup.py
einx/__init__.py
einx/traceback_util.py
einx/tree_util.py
einx/types.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/backend/__init__.py
einx/backend/_dask.py
einx/backend/_jax.py
einx/backend/_mlx.py
einx/backend/_numpy.py
einx/backend/_tensorflow.py
einx/backend/_tinygrad.py
einx/backend/_torch.py
einx/backend/base.py
einx/backend/register.py
einx/experimental/__init__.py
einx/experimental/op/__init__.py
einx/experimental/op/shard.py
einx/expr/__init__.py
einx/expr/solver.py
einx/expr/stage1.py
einx/expr/stage2.py
einx/expr/stage3.py
einx/expr/util.py
einx/nn/__init__.py
einx/nn/equinox.py
einx/nn/flax.py
einx/nn/haiku.py
einx/nn/keras.py
einx/nn/nn.py
einx/nn/torch.py
einx/op/__init__.py
einx/op/arange.py
einx/op/dot.py
einx/op/elementwise.py
einx/op/index.py
einx/op/rearrange.py
einx/op/reduce.py
einx/op/solve.py
einx/op/util.py
einx/op/vmap.py
einx/op/vmap_with_axis.py
einx/tracer/__init__.py
einx/tracer/compile.py
einx/tracer/decorator.py
einx/tracer/input.py
einx/tracer/optimize.py
einx/tracer/tensor.py
einx/tracer/tracer.py
test/test_compare.py
test/test_experimental.py
test/test_nn.py
test/test_shapes.py
test/test_values.py