-
### Description
Hi. I am extending the Pallas paged attention kernel. The case is a MQA. When I run my kernel, I encountered the following error which suggests it is an internal error and I should …
-
Thanks for ZML - I'm pretty eager to play with it. I have successfully run llama3.1 on my CPU, but attempting to run on my GPU(s) causes some issues. The stacktrace I get:
```
F0920 14:59:22.05584…
-
getting issues on torch_xla.core.xla_model. , while installing package also getting errors : "ERROR: Could not find a version that satisfies the requirement torch-xla (from versions: none)
ERROR: No …
-
### Description
Hi. I am extending the Pallas paged attention kernel. The case is a MQA. When I run my kernel, I encountered the following error which suggests it is an internal error and I should re…
-
### Description of the bug:
I'm trying to convert a MobileNetV4 pytorch model to tflite. I get an error regarding the torch_xla library.
### Actual vs expected behavior:
What I get is:
…
-
It seems that the loss is not converging or that we OOM depending on the `XLA_DISABLE_FUNCTIONALIZATION` flag and ZeRO-1.
### System info
```
aws-neuronx-runtime-discovery==2.9
libneuronxla==2.0…
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
Yes
### Source
source
### TensorFlow version
tf 2.17
### Custom code
Yes
### OS platform and distribution
Linux U…
x0w3n updated
8 hours ago
-
Hi,
When I am running following code:
```python
from whisper_jax import FlaxWhisperPipline
import jax.numpy as jnp
pipeline = FlaxWhisperPipline("openai/whisper-large-v2", dtype=jnp.bfloat16, ba…
rrozb updated
1 month ago
-
## ❓ Questions and Help
I follow the code in https://github.com/pytorch/xla/blob/master/contrib/kaggle/distributed-pytorch-xla-basics-with-pjrt.ipynb
But use xmp.spawn(print_device, args=(lock,), …
-
## 🐛 Bug
I am trying to follow https://openxla.org/stablehlo/tutorials/pytorch-export verbatim.
I have taken the official python:3.10 Docker image, and inside it I have installed:
```
--ext…