-
Einstein summation generalizes most tensor operations (product, matrix multiplication, etc).
-
With this issue, I want to kick of the discussion about how to store b-tensor encoding information.
### The b-tensor as a natural extension of the b-vector/value combination
The most natural for…
-
I am trying to recreate my nequip-torch model to nequip-jax. While the number of parameters matches exactly when I create the model, the numerical values do not. SO I am trying to assemble the layers …
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
No
### Source
binary
### TensorFlow version
2.16.1
### Custom code
Yes
### OS platform and dis…
-
Trying to coalesce a few things we've been discussing regarding making transforms more efficient.
## 1. Partial `basis.evaluate`
When calling `basis.evaluate` you should be able to specify which c…
-
### 🐛 Describe the bug
Hi AMD Team,
On MI300X pytorch nightly grouped query attention is running into numeric errors. I have confirmed on H100 that this script does not have numeric errors.
C…
-
### 🐛 Describe the bug
Simple compilation of UNet model works fine, but FSDP-wrapped UNet gets recompiled on every block. In real setup cache-size limit is rapidly reached.
Code:
```
import argp…
-
Is there any guidelines to follow for handling the dynamic shape (e.g. reshaping a dynamic tensor)?
In addition to using the shapeOp + sliceOp, is there any other way to get a specific dim size?
…
-
i don't see this in the docs for the python qutip package, but it would be very handy to have access to subspaces of quantum objects, c.f. this [docs page](https://docs.qojulia.org/quantumsystems/subs…
-
### Describe the issue
With 8 bit file paths there is a high risk that applications won't run in all countries due to many characters not being representable in a char. Using filesystem::path is of c…