-
## ❓ Questions and Help
The code below tries to train an AutoEncoder in Kaggle notebooks. It takes forever to run. I follow the [example notebooks](https://www.kaggle.com/tanlikesmath/the-ultimate-py…
ucals updated
2 years ago
-
### Is there an existing issue for this?
- [X] I have searched the existing issues
### Community Note
* Please vote on this issue by adding a :thumbsup: [reaction](https://blog.github.com…
-
```jax-0.3.7 jaxlib-0.3.7 numpy-1.21.6```
```python
import jax.numpy as np
import jax
jax.config.update('jax_enable_x64', False)
n = 50_000
array = np.eye(n)
vals, vecs = np.linalg.ei…
-
I ran the following command line:
```
python run_squad.py \
--vocab_file=$BERT_LARGE_DIR/vocab.txt \
--bert_config_file=$BERT_LARGE_DIR/bert_config.json \
--init_checkpoint=$BERT_LARGE_DIR/…
-
Hello,I have follow you readme to install the anaconda environment,and it can run with cpu ,but it can not run with gpu ,and in my devices have a nvidia RTX TITAN GPU with 24G momeory ,when I whatever…
-
How do I use TPU
-
getting issues on torch_xla.core.xla_model. , while installing package also getting errors : "ERROR: Could not find a version that satisfies the requirement torch-xla (from versions: none)
ERROR: No …
-
## 🐛 Bug
xm.send_cpu_data_to_device cannot support 2d data and 4d mesh
## To Reproduce
https://colab.research.google.com/drive/1URZVd3q0LUeZ8anzrkoJkehOxLdbG1k8?usp=sharing
```
import numpy…
-
## 🐛 Bug
CPU memory consumption is stable during training.
But memory consumption keep increasing and never decreases at the end of every epochs.
It occurs when I use TPU. (It doesn't occur when I …
-
Hi,
I have problem in JAX installation. I used cluster to run. this is the error :
`Building wheels for collected packages: cvit
Building wheel for cvit (setup.py): started
Building wheel for …