-
## Fix the Op info test for `linalg.lu_solve`
1. Find the line 128 of [test_ops.py](https://github.com/pytorch/xla/blob/master/experimental/torch_xla2/test/test_ops.py) and remove
`linalg.lu_so…
qihqi updated
2 weeks ago
-
## Fix the Op info test for `linalg.matrix_power`
1. Find the line 130 of [test_ops.py](test/test_ops.py) and remove
`linalg.matrix_power` from `skip_list`
2. Run op_info test with `pytest test/te…
qihqi updated
2 weeks ago
-
### Your current environment
The output of `python collect_env.py`
```text
python collect_env.py
Collecting environment information...
2024-09-23 17:57:46.577274: I tensorflow/core/util/po…
-
### 🚀 The feature, motivation and pitch
Hey everyone, I’m Ali, a software engineer from Nuro, and we’ve been working on migrating a lot of vision models from tensorflow to pytorch. One particularly…
-
Hi, thank you so much for sharing this amazing work to everyone!
I am currently testing if MultiNeRF is possible to run on Windows machine, and right now with this command:
```
python -m train --gi…
-
### Description
Run
```
import jax
import jax.numpy as jnp
dtype = jnp.bfloat16
batch_dim = 16
x_shape = (batch_dim, 512, 16, 48)
bias_shape = (batch_dim, 16, 512, 512)
mask_shape = (1, 1, …
-
## 🐛 Bug
The issue looks related to **lifted constants** during `torch.export`, I found a commit https://github.com/pytorch/xla/commit/d8d7e58b78664aff2713e5f25adb3d61c42d44e7 might be related, but…
-
Teorically it you're using transformers, it is possible to train in aws neuron instances (trn1)
With optimun neuron should be possible https://huggingface.co/docs/optimum/main/en/index, https://hug…
-
### Description
Perhaps I am using this function incorrectly, but I get data leaks when using `key_value_seq_lengths`. It appears as though both the `xla` and `cudnn` implementations in jax nightly…
-
## 🐛 Bug
```
python caption.py
WARNING:root:libtpu.so and TPU device found. Setting PJRT_DEVICE=TPU.
Loading checkpoint shards: 100%|██████████████████████████████████████████████████| 2/2 […