Talk:Google JAX

Latest comment: 2 years ago by InfiniteNexus in topic Google JAX
  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