-
### Feature Idea
Support For TPU/XLA Devices
### Existing Solutions
_No response_
### Other
_No response_
-
Running anything in JAX using today's container prints warnings:
```python
import jax.numpy as jnp
2 * jnp.ones(1)
```
Output:
```
2024-10-31 21:29:38.149538: W external/xla/xla/service/gpu/gpu_hlo_sc…
-
We observed good overlap with FSDP + PGLE:
. Turning on and off PGLE makes a big difference here.
…
-
### Your current environment
The output of `python collect_env.py`
```text
Collecting environment information...
PyTorch version: 2.5.1+cu124
Is debug build: False
CUDA used to build PyTorch…
-
## 🐛 Bug
Currently, if I install from nightly using instructions from README.md, and try to run a simple test, I see "undefined symbol: _ZN5torch4lazy13MetricFnValueB5cxx11E" which has "cxx11" in i…
-
The file `ref_2_16x16x16.npy` is shape (2,16**3).
The error seems to be coming from the xmipp ImageHandler, `ih.scaleSplines(data=volume, ...)`.
Probably a version issue.
(flexutils-tensorflo…
-
We are seeing the following errors after a simple `import tensorflow` with TensorFlow/2.15.1-foss-2023a-CUDA-12.1.1:
> 2024-10-25 14:56:51.496918: E external/local_xla/xla/stream_executor/cuda/cuda…
-
I find I get the following warnings about CUDA as I don't have the correct nVidia driver installed.
```
❱ topostats --help
2024-10-11 15:14:19.949079: I external/local_xla/xla/tsl/cuda/cu…
-
### Anything you want to discuss about vllm.
Hi,
We have adapt our harware to pytorch/xla, what should we do to run vllm via pytorch/xla?
-
Hi,
I have problem in JAX installation. I used cluster to run. this is the error :
`Building wheels for collected packages: cvit
Building wheel for cvit (setup.py): started
Building wheel for …