-
The following setup commands:
```
pip install -U "jax[tpu]" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
pip install levanter
```
Given the following error:
```>>> i…
-
Does OpenNMT-tf support training on Cloud TPUs?
-
### 🚀 The feature, motivation, and pitch
trlX uses HuggingFace accelerate under the hood. Accelerate has the capability to leverage Google's TPUs for faster training. I'm interested in supporting trl…
-
Hi, I am running the code and installed the necessary packages as in the requirements.txt. It states pytorch-lightning==1.9.1 and when I run pip show pytorch-lightning
in the terminal it shows
```
…
-
### What happened + What you expected to happen
The author of [this PR](https://github.com/ray-p…
-
Click to expand!
### Issue Type
Bug
### Source
binary
### Tensorflow Version
v2.11.0-0-gd5b57ca9 2.11.0
### Custom Code
No
### OS Platform and Distribution
Linux t1v-n…
-
## 🐛 Bug
Here at AWS we have a single PJRT device plugin for both PyTorch and JAX, and recently we've made implements to our device plugin to make it work better with JAX. I.e. now `PJRT_LoadedExec…
-
## 🐛 Usability Bug
If a user wants to check for a `cuda` device, they call: `torch.cuda.is_available()`
In torch_xla, similar information can be extracted in a few ways. This simple API should n…
-
Is there a way to use a tensor processing unit for acceleration? If not, is this feature going to be added in the future?
-
It's my issue:[https://github.com/tensorflow/tensorflow/issues/37054](url)