Jax Programming

JAX Programming: Harnessing High-Performance Numerical Computing, Automatic Differentiation, and Machine Learning with Python (English Edition)

Independently published

Differentiable Programming for AI Engineers: Practical Workflows with PyTorch, JAX, and Julia

Manning Publications

Deep Learning with JAX                          

Independently published

Software 2.0 Differentiable Programming Essentials: Build Adaptive Algorithms with JAX, PyTorch & Auto‑Diff Techniques

Packt Publishing

Advanced Python Programming: Accelerate your Python programs using proven techniques and design patterns, 2nd Edition

Independently published

Mastering Deep Learning with Python & JAX: Build High-Performance Neural Networks and Scale AI Models to Production (Intelligent Systems Engineering, Band 1)

Independently published

Hands-On Deep Learning with JAX, Flax, and PyTorch: A Practical Guide for Beginners and Beyond

Independently published

Machine Learning for JAX: Building Scalable, Fast, and Flexible AI Models for the Future of Computing

Independently published

Introduction to Google JAX: Your Guide to Understanding JAX 2024 (Google Jax Guide: Chapter one, Band 1)

Independently published

Practical Deep Learning in Python: A Step-by-Step Journey with PyTorch, TensorFlow, Keras, and JAX