make notebook (colab) examples work on TPUs
Open
good first issue
- Dominant language
- Python
- Stars
- 1.1k
- Forks
- 76
- Avg merge
- 2d 21h
- Merged PRs (30d)
- 1
Description
these are the examples on docs/notebooks/
Probably it's necessary to add this line when running on TPUs https://github.com/google/jax#pip-installation-colab-tpu
I think this would be a good first issue
Contributor guide
Assessment
This issue has not been assessed yet.