Talk:Google JAX

This article is rated C-class on Wikipedia's content assessment scale.
It is of interest to the following WikiProjects:

Software: Computing Low‑importance

	This article is within the scope of WikiProject Software, a collaborative effort to improve the coverage of software on Wikipedia. If you would like to participate, please visit the project page, where you can join the discussion and see a list of open tasks.SoftwareWikipedia:WikiProject SoftwareTemplate:WikiProject Softwaresoftware
Low	This article has been rated as Low-importance on the project's importance scale.
	This article is supported by WikiProject Computing.

Google Low‑importance

This article is within the scope of WikiProject Google, a collaborative effort to improve the coverage of Google and related topics on Wikipedia. If you would like to participate, please visit the project page, where you can join the discussion and see a list of open tasks.GoogleWikipedia:WikiProject GoogleTemplate:WikiProject GoogleGoogle

Low This article has been rated as Low-importance on the project's importance scale.

WikiProject Google To-do:

Here are some tasks awaiting attention:

Article requests : Articles for most of the other products listed here and here.
Assess : All articles in the Category:Unknown-importance Google articles and Category:Unassessed Google articles using the project's assessment scale
Expand : Google Mapathon, Google Talkback
Maintain : This WikiProject
Merge : Google mobile services into List of Google products
Stubs : Category:Stub-Class Google articles and Category:Google stubs
Update : List of features in Android and Gmail interface#Product integration. Update logos of Google Marketing Platform products
Other :
- Add more stuff to this to do list if you like! (click here...)
- create:
- Help the Google article for a good article status
- Improve the Outline of Google
- Get more members using :
{{subst:Wikipedia:WikiProject Google/Invite Members}}
- Infobox Images with transparent areas needing a different background color

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

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

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]

@InfiniteNexus What about "JAX (Machine Learning framework)"? Otthorn (talk) 07:37, 31 January 2025 (UTC)[reply]

You can propose a move by following the instructions at WP:RM#CM. InfiniteNexus (talk) 07:45, 31 January 2025 (UTC)[reply]

Requested move 31 January 2025

It has been proposed in this section that Google JAX be renamed and moved to JAX (Machine Learning framework).

A bot will list this discussion on the requested moves current discussions subpage within an hour of this tag being placed. The discussion may be closed 7 days after being opened, if consensus has been reached (see the closing instructions). Please base arguments on article title policy, and keep discussion succinct and civil.

Please use {{subst:requested move}}. Do not use {{requested move/dated}} directly.

Links: current log • target log • direct move

Google JAX → JAX (Machine Learning framework) – If we look at all the papers on arXiv containing "JAX" in their name in the category CS (55 papers), only one use "Google JAX" in it's abstract and another one use "Google's JAX". All the 53 others will cite it has "JAX" like in "EagerPy is a Python framework that lets you write code that automatically works natively with PyTorch, TensorFlow, JAX, and NumPy." for example.

No mention of "Google JAX" in the Github README.

No mention of "Google JAX" on the front page of the documentation. Not even a mention of "Google". It is not even an official Google product: "This is a research project, not an official Google product" from the Github page.

No mention of "Google JAX" in the original paper. Otthorn (talk) 08:24, 31 January 2025 (UTC)[reply]