13 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
68be5b5085 CI: update ruff to v0.6.1 2024-08-27 14:54:11 -07:00
Roy Frostig
371935cc10 update README and several docs to typed RNG keys 2024-08-11 08:09:47 -07: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
d0e21072ec DOC: add brief comments to example in README 2021-09-14 11:47:19 -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
24e920e3f5 change colab.sandbox->colab.research 2021-02-25 10:29:43 -08:00
Skye Wanderman-Milne
85796cc7e3 Fix colab_tpu.setup_tpu import in example Cloud TPU notebooks. 2020-12-14 16:23:20 -08:00
Skye Wanderman-Milne
e802bf9b63 Merge pull request #5173 from jakevdp:conv-elem-type-cleanup
PiperOrigin-RevId: 347085811
2020-12-11 15:27:40 -08:00
Jake Vanderplas
05904faf0f
Change onp/np to np/jnp in docs & notebooks (#3760) 2020-07-15 13:17:38 -07:00
Yash Katariya
236a5881bf
Add try; except to cells that fail and print out the error message so that the notebook can execute end-to-end (#3624) 2020-07-01 08:56:10 -07:00
Jin Dong
a23a6c2a06 remove pip install 2020-04-09 18:16:38 -04:00
Skye Wanderman-Milne
74faef18bf
Rename "NeurIPS 2019 JAX demo" to "JAX demo", as it's been presented elsewhere. (#2372) 2020-03-06 10:32:15 -08:00