Sign-In
Cart
Google JAX Cookbook: Perform machine learning and numerical computing with combined capabilities of TensorFlow and NumPy
Paperback
Publisher Price: $69.99
ISBN13: 9788197950414
Publisher: Gitforgits
Published: Oct 30 2024
Pages: 252
Weight: 0.97
Height: 0.53 Width: 7.50 Depth: 9.25
Language: English
This is the practical, solution-oriented book for every data scientists, machine learning engineers, and AI engineers to utilize the most of Google JAX for efficient and advanced machine learning. It covers essential tasks, troubleshooting scenarios, and optimization techniques to address common challenges encountered while working with JAX across machine learning and numerical computing projects.
The book starts with the move from NumPy to JAX. It introduces the best ways to speed up computations, handle data types, generate random numbers, and perform in-place operations. It then shows you how to use profiling techniques to monitor computation time and device memory, helping you to optimize training and performance. The debugging section provides clear and effective strategies for resolving common runtime issues, including shape mismatches, NaNs, and control flow errors. The book goes on to show you how to master Pytrees for data manipulation, integrate external functions through the Foreign Function Interface (FFI), and utilize advanced serialization and type promotion techniques for stable computations.
If you want to optimize training processes, this book has you covered. It includes recipes for efficient data loading, building custom neural networks, implementing mixed precision, and tracking experiments with Penzai. You'll learn how to visualize model performance and monitor metrics to assess training progress effectively. The recipes in this book tackle real-world scenarios and give users the power to fix issues and fine-tune models quickly.
Key LearningsGet your calculations done faster by moving from NumPy to JAX's optimized framework.
Make your training pipelines more efficient by profiling how long things take and how much memory they use.
Use debugging techniques to fix runtime issues like shape mismatches and numerical instability.
Get to grips with Pytrees for managing complex, nested data structures across various machine learning tasks.
Use JAX's Foreign Function Interface (FFI) to bring in external functions and give your computational capabilities a boost.
Take advantage of mixed-precision training to speed up neural network computations without sacrificing model accuracy.
Keep your experiments on track with Penzai. This lets you reproduce results and monitor key metrics.
Create your own neural networks and optimizers directly in JAX so you have full control of the architecture.
Use serialization techniques to save, load, and transfer models and training checkpoints efficiently.
Table of ContentTransition NumPy to JAX
Profiling Computation and Device Memory
Debugging Runtime Values and Errors
Mastering Pytrees for Data Structures
Exporting and Serialization
Type Promotion Semantics and Mixed Precision
Integrating Foreign Functions (FFI)
Training Neural Networks with JAX
Also in
General Computers
All Data Are Local: Thinking Critically in a Data-Driven Society
Loukissas, Yanni Alexander
Paperback
Comptia Network+ Certmike: Prepare. Practice. Pass the Test! Get Certified!: Exam N10-008
Chapple, Mike
Zacker, Craig
Paperback
Interactive Theorem Proving and Program Development: Coq'art: The Calculus of Inductive Constructions
Castéran, Pierre
Bertot, Yves
Hardcover
R for Data Science: Import, Tidy, Transform, Visualize, and Model Data
Wickham, Hadley
Grolemund, Garrett
Paperback
The Reverse Centaur's Guide to Life After AI: How to Think about Artificial Intelligence--Before It's Too Late
Doctorow, Cory
Paperback
Designing Data-Intensive Applications: The Big Ideas Behind Reliable, Scalable, and Maintainable Systems
Kleppmann, Martin
Riccomini, Chris
Paperback
AI for Good: How Real People Are Using Artificial Intelligence to Fix Things That Matter
Tyrangiel, Josh
Hardcover
If Anyone Builds It, Everyone Dies: Why Superhuman AI Would Kill Us All
Soares, Nate
Yudkowsky, Eliezer
Hardcover
The Year in Tech, 2026: The Insights You Need from Harvard Business Review
Webb, Amy
Review, Harvard Business
Anthony, Scott D.
Paperback
Human Raised: Nurturing Connection, Curiosity & Lifelong Learning in the Age of AI
Suskind, Dana, MD
Hardcover
Hbr's 10 Must Reads on AI (with Bonus Article How to Win with Machine Learning by Ajay Agrawal, Joshua Gans, and AVI Goldfarb)
Davenport, Thomas H.
Iansiti, Marco
Review, Harvard Business
Paperback
The Agi Chronicles: The Inside Story of the Race to Create an Artificial Superintelligence
Roose, Kevin
Hardcover
A Giant Leap: How AI Is Transforming Healthcare and What That Means for Our Future
Wachter, Robert
Hardcover
The AI Con: How to Fight Big Tech's Hype and Create the Future We Want
Hanna, Alex
Bender, Emily M.
Hardcover
Generative Ai: The Insights You Need from Harvard Business Review
Mollick, Ethan
Cremer, David De
Review, Harvard Business
Paperback
CompTIA Network+ Certification All-in-One Exam Guide, Ninth Edition (Exam N10-009)
Weissman, Jonathan S.
Paperback
Fans First: Change The Game, Break the Rules & Create an Unforgettable Experience
Cole, Jesse
Paperback
AI Literacy in Any Class: Preparing Students to Thrive in an AI Future in Any Grade Level or Content Area
Miller, Matt
Paperback
Fundamentals of Data Engineering: Plan and Build Robust Data Systems
Reis, Joe
Housley, Matt
Paperback
Atlas of AI: Power, Politics, and the Planetary Costs of Artificial Intelligence
Crawford, Kate
Paperback
Minecraft: Guide Collection 4-Book Boxed Set (Updated): Survival (Updated), Creative (Updated), Redstone (Updated), Combat
Mojang Ab
The Official Minecraft Team
Hardcover
AI Snake Oil: What Artificial Intelligence Can Do, What It Can't, and How to Tell the Difference
Kapoor, Sayash
Narayanan, Arvind
Paperback
Python Crash Course, 3rd Edition: A Hands-On, Project-Based Introduction to Programming
Matthes, Eric
Paperback
We Are as Gods: A Survival Guide for the Age of Abundance
Diamandis, Peter H.
Kotler, Steven
Hardcover
The Eureka Machine: Why AI Is the Key to Unlocking a New Era of Scientific Discoveries
Socher, Richard
Hardcover
Hands-On Machine Learning with Scikit-Learn and Pytorch: Concepts, Tools, and Techniques to Build Intelligent Systems
Géron, Aurélien
Paperback
AI with Intention: Principles and Action Steps for Teachers and School Leaders
Frontier, Tony
Paperback
Human + Machine, Updated and Expanded: Reimagining Work in the Age of AI
Wilson, H. James
Daugherty, Paul R.
Hardcover
Artificial Intelligence: A Guide for Thinking Humans (with a New Preface)
Mitchell, Melanie
Paperback
The Thinking Machine: Jensen Huang, Nvidia, and the World's Most Coveted Microchip
Witt, Stephen
Hardcover
The Black Swan: Second Edition: The Impact of the Highly Improbable: With a New Section: On Robustness and Fragility
Taleb, Nassim Nicholas
Paperback
Hands-On Large Language Models: Language Understanding and Generation
Alammar, Jay
Grootendorst, Maarten
Paperback
Embedded Systems with ARM Cortex-M Microcontrollers in Assembly Language and C: Fourth Edition
Zhu, Yifeng
Paperback
The Empty Center: Permission, Connection, and What We Build When the Future is Uncertain
Drumwright, Hayes
Hardcover
Tubes: A Journey to the Center of the Internet with a New Introduction by the Author
Blum, Andrew
Paperback
CompTIA Tech+ Certification All-in-One Exam Guide, Third Edition (Exam FC0-U71)
Wempen, Faithe
Paperback
Fundamentals of Software Architecture: A Modern Engineering Approach
Ford, Neal
Richards, Mark
Paperback
AI Projects with Raspberry Pi: High-Performance Artificial Intelligence for Robotics, Security, Home Automation, and Vision
Jepson, Brian
Hattersley, Lucy
Paperback
AP Computer Science a Premium, 13th Edition: Prep Book with 6 Practice Tests + Comprehensive Review + Online Practice (2026)
Carman, Chris
Teukolsky, Roselyn, M. S.
Paperback
Minecraft: Exploded Builds: Medieval Fortress: An Official Mojang Book
The Official Minecraft Team
Mojang Ab
Paperback
Power and Prediction: The Disruptive Economics of Artificial Intelligence
Gans, Joshua
Goldfarb, Avi
Agrawal, Ajay
Hardcover
The Empty Center: Permission, Connection, and What We Build When the Future is Uncertain
Drumwright, Hayes
Paperback
AP Computer Science Principles Premium, 2027: 6 Practice Tests + Comprehensive Review + Online Practice
Reichelson, Seth
Paperback
Love Machines: How Artificial Intelligence Is Transforming Our Relationships
Muldoon, James
Hardcover
Minecraft: Roll for Adventure: The Temple of the Charged Creeper
Forbeck, Matt
Forbeck, Marty
Hardcover
You Look Like a Thing and I Love You: How Artificial Intelligence Works and Why It's Making the World a Weirder Place
Shane, Janelle
Paperback
AI Snake Oil: What Artificial Intelligence Can Do, What It Can't, and How to Tell the Difference
Narayanan, Arvind
Kapoor, Sayash
Hardcover
You to the Power of Two: Redefining Human Potential in the Age of Identic AI
Tapscott, Don
Bradley, Joseph
Hardcover
The Devops Handbook, 2nd Edition: How to Create World-Class Agility, Reliability, & Security in Technology Organizations
Debois, Patrick
Kim, Gene
Humble, Jez
Paperback
The Problem with Personalization: How Advertisers Learned to Make and Break Us from Ancient Times to the AI Age
Turow, Joseph
Hardcover
The AI Ultimatum: Preparing for a World of Intelligent Machines and Radical Transformation
Brown, Steve
Hill, Paul
Paperback
Artificial Intelligence All-In-One for Dummies
Minnick, Chris
Mueller, John Paul
Massaron, Luca
Paperback
The Death of Expertise: The Campaign Against Established Knowledge and Why It Matters
Nichols, Tom
Paperback
Graphic Artists Guild Handbook, 17th Edition: Pricing & Ethical Guidelines
The Graphic Artists Guild
Paperback
The Fourth Intelligence Revolution: The Future of Espionage and the Battle to Save America
Vinci, Anthony
Hardcover
Designing Machine Learning Systems: An Iterative Process for Production-Ready Applications
Huyen, Chip
Paperback
RHCSA Red Hat Enterprise Linux 10: Training and Exam Preparation Guide (EX200), Fourth Edition
Ghori, Asghar
Paperback
AI for Educators: Learning Strategies, Teacher Efficiencies, and a Vision for an Artificial Intelligence Future
Miller, Matt
Paperback
More Human: How the Power of AI Can Transform the Way You Lead
Carter, Jacqueline
Hougaard, Rasmus
Hardcover
Building Applications with AI Agents: Designing and Implementing Multiagent Systems
Albada, Michael
Paperback
Microsoft 365 Excel All-In-One for Dummies
Alexander, Michael
Kusleika, Dick
Ringstrom, David H.
Paperback
The Technological Republic: Hard Power, Soft Belief, and the Future of the West
Karp, Alexander C.
Zamiska, Nicholas W.
Hardcover
The Coming Wave: Technology, Power, and the Twenty-First Century's Greatest Dilemma
Suleyman, Mustafa
Hardcover
The Experimentation Machine: Finding Product-Market Fit in the Age of AI
Bussgang, Jeffrey J.
Hardcover
Hbr's 10 Must Reads on Artificial Intelligence, Updated and Expanded (Featuring How AI Can Help Managers Think Through Problems by Elisa Farri and Gab
Agrawal, Ajay
Review, Harvard Business
Davenport, Thomas H.
Paperback
What Is Intelligence?: Lessons from AI about Evolution, Computing, and Minds
Aguera Y. Arcas, Blaise
Paperback
Navigeren door het labyrint: Een handleiding voor het beheer van data
Sebastian-Coleman, Laura
Paperback
Comptia A+ Complete Study Guide, 2-Volume Set: Volume 1 Core 1 Exam 220-1201 and Volume 2 Core 2 Exam 220-1202
Docter, Quentin
Buhagiar, Jon
Paperback
Prediction Machines, Updated and Expanded: The Simple Economics of Artificial Intelligence
Gans, Joshua
Goldfarb, Avi
Agrawal, Ajay
Hardcover
R for Data Science: Import, Tidy, Transform, Visualize, and Model Data
Grolemund, Garrett
Cetinkaya-Rundel, Mine
Wickham, Hadley
Paperback
Laptops for Seniors in Easy Steps, 9th Edition: Covers All Laptops with the Windows 11 2024 Update
Vandome, Nick
Paperback
On the Future of Species: Authoring Life by Means of Artificial Biological Intelligence
Woolfson, Adrian
Hardcover