Pedro Cuenca commited on
Commit
a11eff5
1 Parent(s): 7cc9b38

Use jax.device_count(), don't assume 8.

Browse files

Former-commit-id: 3c7564038ad4953231aa0f2f5df9359bde28b260

demo/tpu-demo.ipynb ADDED
The diff for this file is too large to render. See raw diff
 
demo/tpu-demo.ipynb.REMOVED.git-id DELETED
@@ -1 +0,0 @@
1
- d3a1b42a490a199c45cd8f75d6be163909866ce1