-
```
(tiny-audio-diffusion) yuki@yuki tiny-audio-diffusion % python train.py exp=drum_diffusion trainer.gpus=1 datamodule.dataset.path=/Users/yuki/Downloads/tiny-audio-diffusion/samples
GPU available…
-
Hi I've been setting up big vision on a v4-32 TPU pod and I run into this error whenever I call `u.put_cpu`
jaxlib.xla_extension.XlaRuntimeError: INVALID_ARGUMENT: Cannot copy array to non-addressa…
-
### Description
Hi there, getting a strange and uninformative error when running on TPU that I don't get when running locally, was hoping someone here could explain the error :thinking:
```
jaxl…
-
I'm an ML Framework ([GoMLX, for Go language](https://github.com/gomlx/gomlx)) developer using XLA (XlaBuilder for now, but at some point changing to PJRT), and I wanted to run some of my trainers in …
-
## 📚 Documentation
Using PyTorch/OpenXLA select a model, and try to get it training and running on Cloud TPUs, create a tutorial on how you went about doing it.
-
### Description
I have a pallas kernel that's dying with a cryptic CHECK failure:
```
Check failed: 4
-
### Bambu Studio Version
1.8.4.51
### Where is the application from?
Bambu Lab Official website
### OS version
Windows 11
### Additional system information
_No response_
### Printer
Bambu A1
…
-
Hi, first of all thanks for this amazing work 👍.
Is it possible to run the `train_network.py` script on a TPU?
I actually tried but it's not working, I even remove the `.to("cuda")` now I'm not se…
-
## 🐛 Bug
I am attempting to implement custom Pallas kernels locally on a CPU for use with a TPU. I'm attempting to follow the official example [here](https://github.com/pytorch/xla/blob/master/docs…
-
## ❓ Questions and Help
I recived error when try create sqmd mesh on kaggle notebook when flow [Huggingface optimum-tpu](https://github.com/huggingface/optimum-tpu/blob/695ee84d657d9ed2761fcf481685af…