Google has developed a new machine learning "monster" called JAX, which is Autograd and XLA, brought together for high-performance machine learning research.
With its updated version of Autograd, JAX can automatically differentiate native Python and NumPy functions. It can differentiate through loops, branches, recursion, and closures, and it can take derivatives of derivatives of derivatives. It supports reverse-mode differentiation (a.k.a. backpropagation) via grad as well as forward-mode differentiation, and the two can be composed arbitrarily to any order. What’s new is that JAX uses XLA to compile and run your NumPy programs on GPUs and TPUs. See more Google JAX.
-
- Active Topics
-
-
- by Eli 1 day ago Re: What is in Your Mind? View the latest post Replies 689 Views 277401
- by Eli 3 days ago Iran Launches Retaliatory Attack Against Israel, and Israel Retaliates by Attacking Iranian Isfahan Millitary Base View the latest post Replies 28 Views 941
- by Eli 5 days ago All in One: YouTube, TED, X, Facebook and Instagram Reels, Videos, Images and Text Posts View the latest post Replies 319 Views 12343
- by Eli 1 week ago Python Packages for Scientific Computing View the latest post Replies 8 Views 6050
- by Eli 1 week ago Dunia Yetu: Building Tanzania's Digital Future Together View the latest post Replies 5 Views 1888
- by Eli 1 week ago Russia Invades Ukraine View the latest post Replies 646 Views 213812
- by Eli 1 week ago Programmatically Move Files from One Folder to Another View the latest post Replies 6 Views 1451
- by Eli 2 weeks ago Collection of Greatest Christian Hymns of all Times View the latest post Replies 33 Views 46938
- by Eli 2 weeks ago What is Retrieval-Augmented Generation (RAG)? View the latest post Replies 2 Views 369
- by Eli 2 weeks ago Chat With ChatGPT - An Interactive Conversational AI View the latest post Replies 22 Views 27678
-
Google JAX: Autograd and XLA
-
- Similar Topics
- Replies
- Views
- Last post
-
- 0 Replies
- 479 Views
-
Last post by Eli
-
- Information
-
Who is online
Users browsing this forum: No registered users and 4 guests