Talk:Google JAX

Hello, I created this page to have an accessible introduction to JAX, and give some MWEs of the 4 primary functions. I am also just learning JAX, so feel free to fix any mistakes.
 * 1) Welcome

vmap example code
I found the vmap example code online (Apache 2.0 licenced). However, I am not quite sure how everything works. If anyone understands this example well and could add a line or two of explanation that would be great (or replace it altogether with a different simple example and an explanation of that example). Bquast (talk) 09:32, 21 June 2022 (UTC)

Google JAX
Pretty sure this is not the common name among practitioners, or developers. Never mind the disclaimer: "This is a research project, not an official Google product." Please consider moving this article elsewhere. 70.134.63.60 (talk) 07:23, 26 October 2022 (UTC)
 * What do you suggest we title the article? InfiniteNexus (talk) 03:36, 27 October 2022 (UTC)