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
|
68be5b5085
|
CI: update ruff to v0.6.1
|
2024-08-27 14:54:11 -07:00 |
|
Jake VanderPlas
|
35b84402c0
|
Deprecate arr.device_buffer and arr.device_buffers
|
2023-12-06 10:20:29 -08:00 |
|
Yash Katariya
|
f0bc7e0fc6
|
Reverts f0382a5838f4526d21631e804f6fe576bfc3f97e
PiperOrigin-RevId: 587231484
|
2023-12-01 22:06:33 -08:00 |
|
Jake VanderPlas
|
97beb01c43
|
Deprecate the device() method of JAX arrays
|
2023-11-30 11:43:02 -08:00 |
|
Jake VanderPlas
|
0aec40a16f
|
Deprecate arr.device_buffer and arr.device_buffers
|
2023-11-29 15:31:01 -08:00 |
|
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 |
|
Jake VanderPlas
|
e1ff0c1d7a
|
Make colab_gpu.ipynb compatible with newer JAX versions
PiperOrigin-RevId: 509356393
|
2023-02-13 15:56:58 -08:00 |
|
Jake VanderPlas
|
5a0a46cbef
|
Update colab CPU test notebook to be more robust
|
2021-07-13 09:30:56 -07: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
|
e1f2c0463e
|
test: use colab_tpu.setup_tpu() in TPU test notebook
|
2021-04-29 11:13:52 -07:00 |
|
Jake Vanderplas
|
d4c0a8d165
|
Add Colab test notebooks for CPU, GPU, and TPU (#3000)
|
2020-05-08 11:11:42 -07:00 |
|