mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/tests/notebooks
History
Peter Hawkins 6ed66ada0f Delete remote TPU support. 
TPU VMs are the only supported way to use TPUs as of JAX 0.4.0.

PiperOrigin-RevId: 519211267
2023-03-24 12:33:33 -07:00
..
colab_cpu.ipynb
Update colab CPU test notebook to be more robust
2021-07-13 09:30:56 -07:00
colab_gpu.ipynb
Make colab_gpu.ipynb compatible with newer JAX versions
2023-02-13 15:56:58 -08:00