JAX Fundamentals for Modern Machine Learning (Paperback)
Lingua: inglese
Editore: Independently Published, 2026
- Brossura
- Nuovo
Da: CitiRetail, Stevenage, Regno UnitoCitiRetail
Venditore AbeBooks dal 29 giugno 2022
Condizione: Nuovo
EUR 33,97
Quantità: 1 disponibile
Aggiungi al carrelloDescrizione dell’articolo da parte del venditore
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.…
Codice articolo 9798192399736
- Titolo
- JAX Fundamentals for Modern Machine Learning (Paperback)
- Autore
- Vincze Kalman
- Editore
- Independently Published
- Anno di pubblicazione
- 2026
- Condizione
- new
- Rilegatura
- Paperback
- Lingua
- inglese
- 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.
"Riassunto" può appartenere a un’altra edizione di questo titolo.
CitiRetail
Stevenage, Regno Unito
Venditore AbeBooks dal 29 giugno 2022
Tariffe di spedizione da Regno Unito a U.S.A.
| Articolo | Da 7 a 14 giorni lavorativi | Da 7 a 60 giorni lavorativi |
|---|---|---|
| Primo articolo | EUR 43,60 | EUR 43,60 |
Metodi di pagamento
Descrizione dello Store
Online business
Informazioni sull’azienda del venditore
ABC BOOKS LIMITED
10 John Street
London, Regno Unito WC1N 2EB
Condizioni di vendita
Orders can be returned within 30 days of receipt.
Diritto di recesso
Se sei un consumatore puoi recedere dal contratto in conformità con quanto segue. Per Consumatore si intende qualsiasi persona fisica che agisce per scopi estranei alla propria attività commerciale, imprenditoriale, artigianale o professionale.
Informazioni sul diritto di recesso
Diritto legale di recesso
Hai il diritto di recedere dal presente contratto entro 14 giorni senza fornire alcuna motivazione.
Il periodo di recesso scade dopo 14 giorni dal giorno in cui tu o una terza parte, diversa dal vettore e da te indicata, acquisisce il possesso fisico dell'ultimo bene o dell'ultimo lotto o pezzo.
Per esercitare il diritto di recesso, compila e invia elettronicamente una dichiarazione esplicita sul nostro sito Web, alla voce “I miei acquisti” nella sezione “Mio account”. Ti comunicheremo senza indugio una conferma di ricezione di tale recesso su un supporto durevole (ad es. via e-mail).
Per rispettare il termine di recesso, è sufficiente inviare la comunicazione relativa all'esercizio del diritto di recesso prima della scadenza del periodo di recesso stesso.
Effetti del recesso
In caso di recesso dal presente contratto, ti rimborseremo tutti i pagamenti ricevuti, compresi i costi di spedizione (ad eccezione dei costi supplementari derivanti dalla tua eventuale scelta di un tipo di spedizione diverso dal tipo meno costoso di consegna standard da noi offerto).
Potremo effettuare una detrazione dal rimborso per la perdita di valore dei beni forniti, qualora tale perdita sia il risultato di una manipolazione non necessaria da parte tua.
Eseguiremo il rimborso senza indebito ritardo e non oltre 14 giorni dal giorno in cui saremo informati della tua decisione di recedere dal presente contratto.
Il rimborso sarà effettuato utilizzando lo stesso mezzo di pagamento da te usato per la transazione iniziale, salvo che tu non abbia espressamente concordato altrimenti; in ogni caso, non dovrai sostenere alcun costo quale conseguenza di tale rimborso.
Possiamo trattenere il rimborso finché non avremo ricevuto i beni oppure finché non avrai fornito la prova di averli rispediti, a seconda di quale condizione si verifichi per prima.
Dovrai rispedire i beni o consegnarli a CitiRetail, Stevenage, United Kingdom, senza indebito ritardo e, in ogni caso, entro 14 giorni dal giorno in cui ci hai comunicato la tua volontà di recedere dal presente contratto. Il termine è rispettato se rispedisci i beni prima della scadenza del periodo di 14 giorni. I costi diretti della restituzione dei beni saranno a tuo carico. Sei responsabile solo della diminuzione del valore dei beni risultante da una manipolazione diversa da quella necessaria per stabilire la natura, le caratteristiche e il funzionamento dei beni stessi.
Eccezioni al diritto di recesso
Il diritto di recesso non si applica a:
- La fornitura di giornali, periodici o riviste ad eccezione dei contratti di abbonamento; e
- La fornitura di contenuto digitale non fornito su un supporto materiale (ad es. su un CD o DVD), se al momento dell'invio dell'ordine hai accettato l'inizio dell'esecuzione e hai riconosciuto che non avresti potuto recedere una volta iniziata l'esecuzione.
Condizioni di spedizione
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.