Sign-In
Cart
Google JAX Essentials: A quick practical learning of blazing-fast library for machine learning and deep learning projects
Paperback
Publisher Price: $59.99
ISBN13: 9788196288358
Publisher: Ingspark
Published: May 31 2023
Pages: 120
Weight: 0.48
Height: 0.25 Width: 7.50 Depth: 9.25
Language: English
Google JAX Essentials is a comprehensive guide designed for machine learning and deep learning professionals aiming to leverage the power and capabilities of Google's JAX library in their projects. Over the course of eight chapters, this book takes the reader from understanding the challenges of deep learning and numerical computations in the existing frameworks to the essentials of Google JAX, its functionalities, and how to leverage it in real-world machine learning and deep learning projects.
The book starts by emphasizing the importance of numerical computing in ML and DL, demonstrating the limitations of standard libraries like NumPy, and introducing the solution offered by JAX. It then guides the reader through the installation of JAX on different computing environments like CPUs, GPUs, and TPUs, and its integration into existing ML and DL projects. The book details the advanced numerical operations and unique features of JAX, including JIT compilation, automatic differentiation, batched operations, and custom gradients. It illustrates how these features can be employed to write code that is both simpler and faster.
The book also delves into parallel computation, the effective use of the vmap function, and the use of pmap for distributed computing. Lastly, the reader is walked through the practical application of JAX in training different deep learning models, including RNNs, CNNs, and Bayesian models, with an additional focus on performance-tuning strategies for JAX applications.
- Mastering the installation and configuration of JAX on various computing environments.
- Understanding the intricacies of JAX's advanced numerical operations.
- Harnessing the power of JIT compilation in JAX for accelerated computations.
- Implementing batched operations using the vmap function for efficient processing.
- Leveraging automatic differentiation and custom gradients in JAX.
- Proficiency in using the pmap function for distributed computing in JAX.
- Training different types of deep learning models using JAX.
- Applying performance tuning strategies to maximize JAX application efficiency.
- Integrating JAX into existing machine learning and deep learning projects.
- Complementing the official JAX documentation with practical, real-world applications.
Table of Content
- Necessity for Google JAX
- Unravelling JAX
- Setting up JAX for Machine Learning and Deep Learning
- JAX for Numerical Computing
- Diving Deeper into Auto Differentiation and Gradients
- Efficient Batch Processing with JAX
- Power of Parallel Computing with JAX
- Training Neural Networks with JAX
Audience
This is must read for machine learning and deep learning professionals to be skilled with the most innovative deep learning library. Knowing Python and experience with machine learning is sufficient is desired to begin with this book
Also from
Wong, Mei
Also in
General Computers
This Program Is Brought to You by . . .: Distributing Television News Online
Braun, Joshua A.
Paperback
The Year in Tech, 2026: The Insights You Need from Harvard Business Review
Review, Harvard Business
Anthony, Scott D.
Webb, Amy
Paperback
If Anyone Builds It, Everyone Dies: Why Superhuman AI Would Kill Us All
Yudkowsky, Eliezer
Soares, Nate
Hardcover
The AI Con: How to Fight Big Tech's Hype and Create the Future We Want
Bender, Emily M.
Hanna, Alex
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)
Iansiti, Marco
Review, Harvard Business
Davenport, Thomas H.
Paperback
Fans First: Change The Game, Break the Rules & Create an Unforgettable Experience
Cole, Jesse
Paperback
Generative Ai: The Insights You Need from Harvard Business Review
Cremer, David De
Review, Harvard Business
Mollick, Ethan
Paperback
The Technological Republic: Hard Power, Soft Belief, and the Future of the West
Zamiska, Nicholas W.
Karp, Alexander C.
Hardcover
Minecraft: Roll for Adventure: The Temple of the Charged Creeper
Forbeck, Matt
Forbeck, Marty
Hardcover
The Experimentation Machine: Finding Product-Market Fit in the Age of AI
Bussgang, Jeffrey J.
Hardcover
Designing Data-Intensive Applications: The Big Ideas Behind Reliable, Scalable, and Maintainable Systems
Kleppmann, Martin
Paperback
AI Snake Oil: What Artificial Intelligence Can Do, What It Can't, and How to Tell the Difference
Kapoor, Sayash
Narayanan, Arvind
Hardcover
Minecraft: Guide Collection 4-Book Boxed Set (Updated): Survival (Updated), Creative (Updated), Redstone (Updated), Combat
The Official Minecraft Team
Mojang Ab
Hardcover
The Black Swan: Second Edition: The Impact of the Highly Improbable: With a New Section: On Robustness and Fragility
Taleb, Nassim Nicholas
Paperback
Atlas of AI: Power, Politics, and the Planetary Costs of Artificial Intelligence
Crawford, Kate
Paperback
Python Crash Course, 3rd Edition: A Hands-On, Project-Based Introduction to Programming
Matthes, Eric
Paperback
The Fourth Intelligence Revolution: The Future of Espionage and the Battle to Save America
Vinci, Anthony
Hardcover
The Thinking Machine: Jensen Huang, Nvidia, and the World's Most Coveted Microchip
Witt, Stephen
Hardcover
You to the Power of Two: Redefining Human Potential in the Age of Identic AI
Tapscott, Don
Bradley, Joseph
Hardcover
Digital Ethics in the Age of AI: Navigating the ethical frontier today and beyond
Mehan, Julie
Paperback
AI for Educators: Learning Strategies, Teacher Efficiencies, and a Vision for an Artificial Intelligence Future
Miller, Matt
Paperback
Artificial Intelligence All-In-One for Dummies
Mueller, John Paul
Massaron, Luca
Minnick, Chris
Paperback
Hands-On Large Language Models: Language Understanding and Generation
Alammar, Jay
Grootendorst, Maarten
Paperback
More Human: How the Power of AI Can Transform the Way You Lead
Hougaard, Rasmus
Carter, Jacqueline
Hardcover
The Coming Wave: Technology, Power, and the Twenty-First Century's Greatest Dilemma
Suleyman, Mustafa
Hardcover
The Podcast Pantheon: 101 Podcasts That Changed How We Listen--From Wtf to Serial
Malin, Sean
Hardcover
AI Made Simple. Results Made Real.: An Executive's Guide to Partnering with the Future
Perley, Kathleen
Paperback
AI Snake Oil: What Artificial Intelligence Can Do, What It Can't, and How to Tell the Difference
Kapoor, Sayash
Narayanan, Arvind
Paperback
Tubes: A Journey to the Center of the Internet with a New Introduction by the Author
Blum, Andrew
Paperback
Artificial Intelligence: A Guide for Thinking Humans (with a New Preface)
Mitchell, Melanie
Paperback
The Cybernetic Society: How Humans and Machines Will Shape the Future Together
Husain, Amir
Hardcover
Beast in the Machine: How Robotics and AI Will Transform Warfare and the Future of Human Conflict
Dougherty, George M.
Hardcover
Microsoft 365 Excel All-In-One for Dummies
Kusleika, Dick
Ringstrom, David H.
Alexander, Michael
Paperback
Laptops for Seniors in Easy Steps, 9th Edition: Covers All Laptops with the Windows 11 2024 Update
Vandome, Nick
Paperback
AP Computer Science Principles Premium, 2026: Prep Book with 6 Practice Tests + Comprehensive Review + Online Practice
Reichelson, Seth
Paperback
What Is Intelligence?: Lessons from AI about Evolution, Computing, and Minds
Aguera Y. Arcas, Blaise
Paperback
The Death of Expertise: The Campaign Against Established Knowledge and Why It Matters
Nichols, Tom
Paperback
Minecraft: Exploded Builds: Medieval Fortress: An Official Mojang Book
Mojang Ab
The Official Minecraft Team
Paperback
RHCSA Red Hat Enterprise Linux 9: Training and Exam Preparation Guide (EX200), Third Edition
Ghori, Asghar
Paperback
The Emergent Mind: How Intelligence Arises in People and Machines
Suri, Gaurav
McClelland, Jay
Hardcover
Embedded Systems with ARM Cortex-M Microcontrollers in Assembly Language and C: Fourth Edition
Zhu, Yifeng
Paperback
AI Valley: Microsoft, Google, and the Trillion-Dollar Race to Cash in on Artificial Intelligence
Rivlin, Gary
Hardcover
The Year in Tech, 2025: The Insights You Need from Harvard Business Review
Review, Harvard Business
Webb, Amy
Farri, Elisa
Paperback
Fundamentals of Data Engineering: Plan and Build Robust Data Systems
Reis, Joe
Housley, Matt
Paperback
The Devops Handbook, 2nd Edition: How to Create World-Class Agility, Reliability, & Security in Technology Organizations
Humble, Jez
Debois, Patrick
Kim, Gene
Paperback
The Magic of Code: How Digital Language Created and Connects Our World--And Shapes Our Future
Arbesman, Samuel
Hardcover
Mindmasters: The Data-Driven Science of Predicting and Changing Human Behavior
Matz, Sandra
Hardcover
Designing Machine Learning Systems: An Iterative Process for Production-Ready Applications
Huyen, Chip
Paperback
Building Applications with AI Agents: Designing and Implementing Multiagent Systems
Albada, Michael
Paperback
The AI Ultimatum: Preparing for a World of Intelligent Machines and Radical Transformation
Brown, Steve
Hill, Paul
Paperback
Me, My Customer, and AI: The New Rules of Entrepreneurship
Thorne, Nicholas
Werdelin, Henrik
Hardcover
R for Data Science: Import, Tidy, Transform, Visualize, and Model Data
Cetinkaya-Rundel, Mine
Wickham, Hadley
Grolemund, Garrett
Paperback
Fundamentals of Software Architecture: A Modern Engineering Approach
Richards, Mark
Ford, Neal
Paperback
Prompt Engineering for Generative AI: Future-Proof Inputs for Reliable AI Outputs
Phoenix, James
Taylor, Mike
Paperback
Practical Charts: The Essential Guide to Creating Clear, Compelling Charts for Reports and Presentations
Desbarats, Nicholas P.
Paperback
Graphic Artists Guild Handbook, 17th Edition: Pricing & Ethical Guidelines
The Graphic Artists Guild
Paperback
Hands-On Machine Learning with Scikit-Learn and Pytorch: Concepts, Tools, and Techniques to Build Intelligent Systems
Géron, Aurélien
Paperback
AI for the Authentic Leader: How to Communicate More Effectively Without Losing Your Humanity
Shapira, Allison
Hardcover
Building AI-Powered Products: The Essential Guide to AI and Genai Product Management
Nika, Marily
Paperback
AI for Life: 100+ Ways to Use Artificial Intelligence to Make Your Life Easier, More Productive...and More Fun!
Quillian, Celia
Paperback
Verified: How to Think Straight, Get Duped Less, and Make Better Decisions about What to Believe Online
Wineburg, Sam
Caulfield, Mike
Paperback
Mastering Video Content Creation: A Practical Guide to Social Media Growth with Expertly Shot and Edited Posts
Espejo, Justin
Paperback
AI for Business: The Beginner's Fast Track to ChatGPT for Productivity, Profit, and Growth (2 books in 1)
Grant, Russel
Paperback
The Chaos Machine: The Inside Story of How Social Media Rewired Our Minds and Our World
Fisher, Max
Paperback
Isc2 Cissp Certified Information Systems Security Professional Official Study Guide & Practice Tests Bundle
Chapple, Mike
Stewart, James Michael
Gibson, Darril
Paperback
The Mechanic and the Luddite: A Ruthless Criticism of Technology and Capitalism
Sadowski, Jathan
Paperback
Software Architecture: The Hard Parts: Modern Trade-Off Analyses for Distributed Architectures
Sadalage, Pramod
Ford, Neal
Richards, Mark
Paperback
Rewiring Your Mind for AI: How to Think, Work, and Thrive in the Age of Intelligence
Wood, David a.
Paperback
AI with Intention: Principles and Action Steps for Teachers and School Leaders
Frontier, Tony
Paperback
Rewiring Democracy: How AI Will Transform Our Politics, Government, and Citizenship
Schneier, Bruce
Sanders, Nathan E.
Hardcover
Prompt Engineering for Llms: The Art and Science of Building Large Language Model-Based Applications
Berryman, John
Ziegler, Albert
Paperback
Brave New Words: How AI Will Revolutionize Education (and Why That's a Good Thing)
Khan, Salman
Hardcover
Prediction Machines, Updated and Expanded: The Simple Economics of Artificial Intelligence
Agrawal, Ajay
Gans, Joshua
Goldfarb, Avi
Hardcover
Cloud Finops: Collaborative, Real-Time Cloud Value Decision Making
Storment, J. R.
Fuller, Mike
Paperback
How To Think With AI: A Simple Guide to Boost Your Brain Power, Creativity, and Performance
McCauley, Alison
Hardcover
Algorithms to Live by: The Computer Science of Human Decisions
Griffiths, Tom
Christian, Brian
Paperback