JAX Fundamentals for Modern Machine Learning (Paperback)
Idioma: inglés
Editorial: Independently Published, 2026
- Tapa blanda
- Nuevo

Librería: CitiRetail, Stevenage, Reino UnidoCitiRetail
Vendedor de IberLibro desde 29 de junio de 2022
Condición: Nuevo
EUR 33,97
Cantidad disponible: 1 disponible
Añadir al carritoDescripción del artículo del vendedor
Paperback. Learn JAX from the ground up and discover how its most powerful features fit together in a practical machine learning workflow.JAX brings together NumPy-style array computing, automatic differentiation, compilation, vectorization, and hardware acceleration. These capabilities make it an exciting tool for modern machine learning, but they can also make JAX feel difficult to approach when you are just getting started.JAX Fundamentals for Modern Machine Learning provides a clear, practical introduction designed for Python users who want to understand JAX without being overwhelmed by research-level examples or unnecessary complexity.Instead of treating JAX as a collection of isolated functions, this book teaches the fundamentals progressively. You will begin with arrays and JAX-friendly Python, then build your understanding of gradients, JIT compilation, vectorization, neural networks, and accelerated execution as each concept becomes useful.Throughout the book, you will gradually build a handwritten digit classifier with a neural network, giving every major JAX concept a practical purpose.Inside this book, you will learn how to: Create, reshape, inspect, and manipulate JAX arraysUnderstand shapes, data types, broadcasting, and immutable updatesWrite pure functions and manage state in a JAX-friendly wayWork correctly with JAX random number keysOrganize model parameters using PyTreesCalculate derivatives and gradients with jax.gradCompute values and gradients together with jax.value_and_gradSpeed up numerical functions using jax.jitUnderstand tracing, recompilation, and common JIT errorsVectorize computations and batches with jax.vmapCombine vmap, grad, and jit in practical workflowsBuild a neural network from basic JAX operationsCreate loss and accuracy functionsCompute gradients and update model parameters during trainingBuild and run a compiled training stepEvaluate a trained classifier and make predictionsSave and reload learned model parametersUnderstand CPU, GPU, and TPU execution in JAXMeasure performance and work with asynchronous executionDiagnose common JAX errors involving shapes, tracers, devices, randomness, and trainingThe examples stay focused on the skills you actually need to understand JAX. You will not be asked to memorize a large framework or copy a finished machine learning system without understanding how it works.By the end of the book, you will have built a complete beginner-friendly machine learning project while developing a practical understanding of the JAX tools that make modern numerical computing fast, composable, and powerful.Whether you are coming from Python, NumPy, machine learning, or another numerical computing library, this book will give you the foundation you need to start using JAX with confidence. This item is printed on demand. Shipping may be from our UK warehouse or from our Australian or US warehouses, depending on stock availability.…
N° de ref. del artículo 9798192399736
- Título
- JAX Fundamentals for Modern Machine Learning (Paperback)
- Autor
- Vincze Kalman
- Editorial
- Independently Published
- Año de publicación
- 2026
- Estado
- new
- Encuadernación
- Paperback
- Idioma
- inglés
- ISBN 13
- 9798192399736
Learn JAX from the ground up and discover how its most powerful features fit together in a practical machine learning workflow.
JAX brings together NumPy-style array computing, automatic differentiation, compilation, vectorization, and hardware acceleration. These capabilities make it an exciting tool for modern machine learning, but they can also make JAX feel difficult to approach when you are just getting started.
JAX Fundamentals for Modern Machine Learning provides a clear, practical introduction designed for Python users who want to understand JAX without being overwhelmed by research-level examples or unnecessary complexity.
Instead of treating JAX as a collection of isolated functions, this book teaches the fundamentals progressively. You will begin with arrays and JAX-friendly Python, then build your understanding of gradients, JIT compilation, vectorization, neural networks, and accelerated execution as each concept becomes useful.
Throughout the book, you will gradually build a handwritten digit classifier with a neural network, giving every major JAX concept a practical purpose.
Inside this book, you will learn how to:-
Create, reshape, inspect, and manipulate JAX arrays
-
Understand shapes, data types, broadcasting, and immutable updates
-
Write pure functions and manage state in a JAX-friendly way
-
Work correctly with JAX random number keys
-
Organize model parameters using PyTrees
-
Calculate derivatives and gradients with jax.grad
-
Compute values and gradients together with jax.value_and_grad
-
Speed up numerical functions using jax.jit
-
Understand tracing, recompilation, and common JIT errors
-
Vectorize computations and batches with jax.vmap
-
Combine vmap, grad, and jit in practical workflows
-
Build a neural network from basic JAX operations
-
Create loss and accuracy functions
-
Compute gradients and update model parameters during training
-
Build and run a compiled training step
-
Evaluate a trained classifier and make predictions
-
Save and reload learned model parameters
-
Understand CPU, GPU, and TPU execution in JAX
-
Measure performance and work with asynchronous execution
-
Diagnose common JAX errors involving shapes, tracers, devices, randomness, and training
The examples stay focused on the skills you actually need to understand JAX. You will not be asked to memorize a large framework or copy a finished machine learning system without understanding how it works.
By the end of the book, you will have built a complete beginner-friendly machine learning project while developing a practical understanding of the JAX tools that make modern numerical computing fast, composable, and powerful.
Whether you are coming from Python, NumPy, machine learning, or another numerical computing library, this book will give you the foundation you need to start using JAX with confidence.
“Sinopsis” puede pertenecer a otra edición de este título.
CitiRetail
Stevenage, Reino Unido
Vendedor de IberLibro desde 29 de junio de 2022
Tarifas de envío de Reino Unido a Estados Unidos de America
| Artículo | De 7 a 14 días hábiles | De 7 a 60 días hábiles |
|---|---|---|
| Primer artículo | EUR 43,60 | EUR 43,60 |
Métodos de pago
Descripción de la tienda
Online business
Información empresarial del vendedor
ABC BOOKS LIMITED
10 John Street
London, Reino Unido WC1N 2EB
Condiciones de venta
Orders can be returned within 30 days of receipt.
Derecho al desistimiento
Si es un consumidor, puede rescindir el contrato de acuerdo con lo siguiente. Por consumidor se entiende cualquier persona física que actúe con fines ajenos a su actividad comercial, empresarial, oficio o profesión.
Información sobre el derecho de desistimiento
Derecho legal de desistimiento
Tiene derecho a rescindir este contrato en un plazo de 14 días sin dar ningún motivo.
El periodo de desistimiento vencerá a los 14 días desde que usted, o un tercero que no sea el transportista e indicado por usted, adquiera la posesión física del último bien o del último lote o pieza.
Para ejercer el derecho de desistimiento, complete de forma electrónica y envíe una declaración clara en nuestro sitio web, desde "Mis compras" en "Mi cuenta". Le enviaremos sin demora un acuse de recibo de dicho desistimiento a través de un soporte duradero (por ejemplo, por correo electrónico).
Para cumplir con el plazo de desistimiento, basta con que envíe su comunicación relativa al ejercicio del derecho de desistimiento antes de que venza el periodo de desistimiento.
Efectos del desistimiento
Si rescinde este contrato, le reembolsaremos todos los pagos que hayamos recibido de usted, incluidos los gastos de envío (excepto los gastos adicionales que surjan si elige un tipo de envío que no sea el tipo de envío estándar más económico que ofrecemos).
Podemos hacer una deducción del reembolso por la pérdida de valor de cualquier bien suministrado, si la pérdida es el resultado de una manipulación innecesaria por su parte.
Efectuaremos el reembolso sin demoras indebidas y, a más tardar, 14 días después de que se nos informe de su decisión de rescindir este contrato.
Efectuaremos el reembolso utilizando el mismo medio de pago que utilizó para la transacción inicial, a menos que haya acordado expresamente lo contrario; en cualquier caso, no incurrirá en ningún cargo como resultado de dicho reembolso.
Podremos retener el reembolso hasta que hayamos recibido los bienes o hasta que nos haya presentado una prueba de que los ha devuelto, lo que ocurra primero.
Deberá devolver los bienes o entregarlos a CitiRetail, Stevenage, United Kingdom, sin demoras indebidas y, en cualquier caso, en un plazo máximo de 14 días a partir del día en que nos comunique su desistimiento del presente contrato. El plazo se cumple si devuelve la mercancía antes de que venza el periodo de 14 días. Tendrá que asumir los gastos directos de devolución de los bienes. Usted solo es responsable de la disminución del valor de los bienes como resultado de una manipulación distinta a la necesaria para establecer la naturaleza, las características y el funcionamiento de los bienes.
Excepciones al derecho de desistimiento
El derecho de desistimiento no se aplica a lo siguiente:
- La entrega de periódicos, diarios o revistas, con la excepción de los contratos de suscripción; y
- El suministro de contenido digital que no se proporcione en un soporte tangible (por ejemplo, en un CD o DVD) si, al hacer el pedido, aceptó que podíamos empezar a entregarlo y que no podría desistir una vez iniciada la entrega.
Condiciones de envío
Please note that titles are dispatched from our US, Canadian or Australian warehouses. Delivery times specified in shipping terms. Orders ship within 2 business days. Delivery to your door then takes 7-14 days.