-
Right now, the program crashes whenever you make a tensor on an XLA device and call a `LazyTensorBarrier()` without setting `wait` to `true`. This can be bypassed by using `LazyTensorBarrier(wait: tru…
-
When we create a `BigQueryClient` instance in the scope of TPUStrategy, i got following error.
```
NotFoundError: 'IO>BigQueryClient' is neither a type of a primitive operation nor a
name of a f…
-
Using `to_tf_dataset` to create a dataset and then putting it in `model.fit` results in an internal error on TPUs. I've only tried on Colab and Kaggle TPUs, not GCP TPUs.
## Steps to reproduce th…
-
### Description
Hello,
I'm encountering an issue that when converting negative float16 values to int8 in pallas kernel, the conversion will fall incorrectly to zero value instead of a negative int8 …
-
### Affected Resource(s)
* google_container_cluster
* google_container_node_pool
According to this page, https://cloud.google.com/kubernetes-engine/docs/reference/rest/v1/projects.locations…
-
Is it possible to train Albert from scratch in another language using a TPU v3 (128Gb)?
Could you give an estimated training time? Days, weeks, months?
What is a reasonable corpus size? 1B words…
-
Hi,
So I have been trying to make use of the composite optimizer in the tf-recommenders package for an object detection project. There are issues that result from training the model on the TPU, th…
-
When I run the entire code my estimator shuts down automatically, but when running a certain function for prediction the estimator will be initiated but I don't see the shutdown info about the TPU. Ho…
-
I'm trying to run faster rcnn resnet inception v2 on a cloud TPU instance, I know that there is no config in the object detection API that is compatible with TPUs, but I am getting always the followin…
-
Currently [supported platforms](https://jax.readthedocs.io/en/latest/installation.html#supported-platforms) are CPUs, TPUs, and Nvidia GPUs, with AMD GPUs and the Metal API for MacOS being experimenta…