-
I followed the installation instructions but wasn't able to find the command of the-info.
If I did `pip list | grep tpu-info` it shows `tpu-info 0.1.0`. But the command `tpu-info` is still not foun…
-
My TPU (95A) filament kinks here
![image](https://github.com/user-attachments/assets/a0505a58-2d82-476f-938e-dff24189c16c)
![image](https://github.com/user-attachments/assets/3f2f49cb-ae52-4e3d-…
-
### Description
I am trying to fine-tune Gemma 2 on TPU and got the following error:
```
Traceback (most recent call last):
File "/usr/local/lib/python3.10/dist-packages/jax/_src/compiler.py", l…
-
## 🐛 Bug
## To Reproduce
Steps to reproduce the behavior:
0.
```
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
pip install 'torch…
-
How do you install the Custom Models for TPU? The Coral Model doesn't have a custom model folder and I don't think it uses YOLO as far as I'm aware at least.
-
Hi, Ive printed my portal and its my first time using TPU, the shape looks ok, but when inserting say the V2_Connector_Blocker I can see there is a small gap there when inserted because the flexible p…
-
PGX on TPUs seems to be slower than CPUs.
With a TPU v3-8, PGX is only achieving 1638 steps / sec on the game of chess.
**Minimal Reproducible Example**
PGX CPU vs TPU Test (512 env) (with shard…
wtedw updated
2 weeks ago
-
us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.1.0_3.10_cuda_12.1
-
### Describe the bug
wandb version: 0.17.7
pytorch lightning version: 2.4
I am using `WandbLogger` following this [guide](https://docs.wandb.ai/guides/integrations/lightning) on TPU.
When I use …
-
### What happened + What you expected to happen
Somehow all TPU slice workers are getting the special TPU-{type}-head resource, despite the fact that it should only be going to the actual head. I'm v…