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 8 hours ago All in One: YouTube, TED, X, Facebook and Instagram Reels, Videos, Images and Text Posts View the latest post Replies 326 Views 31815
- by Eli 1 day ago Russia Invades Ukraine View the latest post Replies 662 Views 232789
- by Eli 1 day ago PySpark for Large Data Processing View the latest post Replies 1 Views 4373
- by Eli 3 days ago Re: What is in Your Mind? View the latest post Replies 700 Views 297467
- by Eli 6 days ago President Museveni's Speech During International Development Association (IDA) Summit View the latest post Replies 1 Views 252
- by Eli 6 days ago From Simple Linear Regression Analysis to Covariance & Correlation to Independent Determinant, and R-Squared View the latest post Replies 11 Views 24716
- by Eli 1 week ago Collection of Greatest Christian Hymns of all Times View the latest post Replies 34 Views 65152
- by Eli 1 week ago Pondering Big Cosmology Questions Through Lectures and Dialogues View the latest post Replies 34 Views 57943
- by Eli 1 week ago Programmatically Manipulate Files: Renaming, Reading, Writing, Deleting, and Moving Files Between Folders View the latest post Replies 7 Views 13677
- by Eli 2 weeks ago Iran Launches Retaliatory Attack Against Israel, and Israel Retaliates by Attacking Iranian Isfahan Millitary Base View the latest post Replies 28 Views 14989
-
Google JAX: Autograd and XLA
-
- Similar Topics
- Replies
- Views
- Last post
-
- 0 Replies
- 656 Views
-
Last post by Eli
-
- Information
-
Who is online
Users browsing this forum: No registered users and 3 guests