Talk:Google JAX

Page contents not supported in other languages.
From Wikipedia, the free encyclopedia
  1. Welcome

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.

vmap example code[edit]

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)[reply]

Google JAX[edit]

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)[reply]

What do you suggest we title the article? InfiniteNexus (talk) 03:36, 27 October 2022 (UTC)[reply]