Google JAX: Autograd and XLA

Includes All Types of AI, Data Science, Data Mining, Machine Learning, Deep Learning, Big Data, Internet of Things & BlockChain Technology
Post Reply
User avatar
Eli
Senior Expert Member
Reactions: 183
Posts: 5334
Joined: 9 years ago
Location: Tanzania
Has thanked: 75 times
Been thanked: 88 times
Contact:
Contact Eli

#1

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.
0
TSSFL -- A Creative Journey Towards Infinite Possibilities!
Top
Post Reply
  • Similar Topics
    Replies
    Views
    Last post

Return to “AI, Generative AI, Artificial General Intelligence, BlockChain, IoT”

  • Information

  • Who is online

    Users browsing this forum: No registered users and 6 guests