Spaces:

Hazem
/

Fac256xc

Runtime error

App Files Files Community

Fac256xc / demo /tpu-demo.ipynb

Pedro Cuenca

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

a11eff5 almost 4 years ago

4.44 MB