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
26,443 Commits 504 Branches 263 Tags
Commit Graph

5 Commits

Author SHA1 Message Date
Michael Hudgins
d4d1518c3d Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax 
PiperOrigin-RevId: 676843138
 2024-09-20 07:52:33 -07:00
Jake VanderPlas
97beb01c43 Deprecate the device() method of JAX arrays 2023-11-30 11:43:02 -08:00
Jake VanderPlas
e1ff0c1d7a Make colab_gpu.ipynb compatible with newer JAX versions 
PiperOrigin-RevId: 509356393
 2023-02-13 15:56:58 -08:00
George Necula
6a48c60a72 Rename master to main in embedded links. 
Tried to avoid the change on external links to repos that
have not yet renamed master.
 2021-06-18 10:00:01 +03:00
Jake Vanderplas
d4c0a8d165
Add Colab test notebooks for CPU, GPU, and TPU (#3000) 2020-05-08 11:11:42 -07:00