-
I'm seeing opt_einsum generate suboptimal schedules for einsum graphs of small treewidth
For instance, family of Apollonian networks are treewidth=3 graphs where optimal decomposition is discovered…
-
I was benchmarking effect of reusing `einsum` contraction path with `opt_einsum` package which is used internally in JAX (see [code][gist]). There are three cases in the benchmark. The first one cas…
-
Currently, `torch.einsum` is computed by performing tensor contractions from left to right and calling `torch.bmm` for each contraction, unless there are no dimensions to be summed out in which case `…
-
```
INFO: pip is looking at multiple versions of opt-einsum to determine which version is compatible with other requirements. This could take a while.
Collecting opt-einsum>=2.3.2
Downloading https:…
-
## Purpose
The idea behind this collection is to provide an index to easily navigate all currently open ndarray's issues which are **immediately** actionable.
This is meant to be a good starting p…
-
Sorry to post so many issues in a row, but it seems like my use-case is encountering some edge problems.
Now that #192 is solved, large copy nodes can be instantiated, but there are some other limi…
-
### Description
```
>>> import jax.numpy as jnp
>>> a = jnp.ones((2,3,4))
Platform 'METAL' is experimental and not all JAX functionality may be correctly supported!
2024-03-06 22:43:05.128623: W …
dlwh updated
5 months ago
-
Since many contractors are based on `opt_einsum` it might be useful to add support for its [path reuse functionality](https://optimized-einsum.readthedocs.io/en/latest/reusing_paths.html). This would …
-
For the following code
```
from opt_einsum import contract
import numpy as np
import opt_einsum as oe
import timeit
dim = 30
I = np.random.rand(dim, dim, dim, dim)
C = np.random.rand(dim, di…
-
when I train the model using batch_size=2, the error is below:
please help me, thank you very much!
Traceback (most recent call last):
File "main.py", line 251, in
main(args_)
File "m…