Название: JAX in Action (MEAP v3)
Автор: Grigory Sapunov
Издательство: Manning Publications
Год: 2022
Страниц: 147
Язык: английский
Формат: pdf (true)
Размер: MB
Accelerate deep learning and other number-intensive tasks with JAX, Google’s awesome high-performance numerical computing library. The JAX numerical computing library tackles the core performance challenges at the heart of Deep Learning and other scientific computing tasks. By combining Google’s Accelerated Linear Algebra platform (XLA) with a hyper-optimized version of NumPy and a variety of other high-performance features, JAX delivers a huge performance boost in low-level computations and transformations. You’ll learn how to use JAX’s ecosystem of high-level libraries and modules, and also how to combine TensorFlow and PyTorch with JAX for data loading and deployment. The JAX Python mathematics library is used by many successful deep learning organizations, including Google’s groundbreaking DeepMind team. This exciting newcomer already boasts an amazing ecosystem of tools including high-level deep learning libraries Flax by Google, Haiku by DeepMind, gradient processing and optimization libraries, libraries for evolutionary computations, federated learning, and much more! JAX brings a functional programming mindset to Python deep learning, letting you improve your composability and parallelization in a cluster.