27 Commits

Author SHA1 Message Date
Skye Wanderman-Milne
8a65ff8c25 Update cloud_tpu_colabs README to reflect Cloud TPU VM announcement. 2021-06-02 16:55:52 -07:00
Jake VanderPlas
24e920e3f5 change colab.sandbox->colab.research 2021-02-25 10:29:43 -08:00
George Necula
d6247defd6
Update Pmap_Cookbook.ipynb 2021-01-31 22:19:13 +02:00
George Necula
580a8f54cf
Update Pmap_Cookbook.ipynb
I am doing this mostly to trigger another run of CI; there was a problem with our CI runs and this PR appears to be failing tests.
2021-01-31 16:23:39 +02:00
mat kelcey
4e48023538
update out of date comment in Pmap_Cookbook.ipynb
update comment regarding `pmap` not support `in_axes` because it does now.
2021-01-31 09:07:55 +11:00
James Bradbury
1e2c5551db
fix dropped import 2021-01-13 11:07:50 -08:00
Skye Wanderman-Milne
85796cc7e3 Fix colab_tpu.setup_tpu import in example Cloud TPU notebooks. 2020-12-14 16:23:20 -08:00
Skye Wanderman-Milne
e802bf9b63 Merge pull request #5173 from jakevdp:conv-elem-type-cleanup
PiperOrigin-RevId: 347085811
2020-12-11 15:27:40 -08:00
Skye Wanderman-Milne
9f4015d41a Add NeurIPS 2020 demo notebook. 2020-12-07 20:03:29 +00:00
Skye Wanderman-Milne
a6b0709873 Demo notebook placeholder 2020-12-04 15:46:51 -08:00
Jake Vanderplas
05904faf0f
Change onp/np to np/jnp in docs & notebooks (#3760) 2020-07-15 13:17:38 -07:00
Yash Katariya
aaf0c4e36b
Install scikit-image and add try;except around from google.colab import files import statement as google.colab is not available locally and in kokoro (#3626) 2020-07-01 11:41:24 -07:00
Yash Katariya
236a5881bf
Add try; except to cells that fail and print out the error message so that the notebook can execute end-to-end (#3624) 2020-07-01 08:56:10 -07:00
Matthew Johnson
e9ce700b06
update tpu readme not to pin jax / jaxlib versions (#3467) 2020-06-16 15:48:01 -07:00
Jin Dong
a23a6c2a06 remove pip install 2020-04-09 18:16:38 -04:00
Skye Wanderman-Milne
74faef18bf
Rename "NeurIPS 2019 JAX demo" to "JAX demo", as it's been presented elsewhere. (#2372) 2020-03-06 10:32:15 -08:00
Ron Weiss
a473e5b6bb
Reference deduped precision issue. (#2319) 2020-02-27 10:20:01 -05:00
Skye Wanderman-Milne
f6d483739a
Shrink nested pmap image in demo notebook. (#2293) 2020-02-22 23:10:13 -08:00
Skye Wanderman-Milne
7885b1d034
Add nested pmap image to demo notebook. (#2292) 2020-02-22 23:04:34 -08:00
Skye Wanderman-Milne
5e77789afe Update Cloud TPU email address to jax-tpu@googlegroups.com 2020-02-11 13:42:11 -08:00
Matthew Johnson
9afa2c6b69 fix broken link to trax, fixes #1974 2020-01-10 20:44:24 -08:00
Skye Wanderman-Milne
a6da7e8bf9
Link to "Reporting issues" at top of Cloud TPU README. 2019-12-16 09:58:30 -08:00
Matthew Johnson
c952ed4f5f
update readme to clarify Cloud TPU preview status 2019-12-16 09:32:55 -08:00
Matthew Johnson
6ba39f763d tweak demo 2019-12-14 10:33:37 -08:00
Matthew Johnson
5c800367d1 mention cloud tpus in readme 2019-12-14 10:32:04 -08:00
Skye Wanderman-Milne
1efe648d35 Add demo colab. 2019-12-13 16:17:58 -08:00
Skye Wanderman-Milne
1f88c831bb
Add Cloud TPU Colabs landing page. 2019-12-13 15:48:32 -08:00