Skip to main content

Search

Libraries Admin

Home
Textbook Content
Part 1 Mathematical Foundations for Machine Learning
Part 2 Python Ecosystem and Classical Machine Learning
Part 3 Deep Learning Fundamentals
Part 4 Modern Architectures and Large Language Models
Part 5 Generative Models
Part 6 Multimodal AI and Vision-Language Models
- Ch 21: Vision-Language Foundations
- Ch 22: Text-to-Image and Beyond
Part 7 Applied AI Systems
Appendices
Appendix A: Environment Setup Guide
Appendix B: GPU Computing Fundamentals
Appendix C: Cloud Platform Quick Start
Appendix D: Mathematics Reference
Appendix E: Library Quick Reference
Resources
Library Reference
Learning Paths

Home / Libraries / JAX

JAX

Deep Learning

Composable transformations of Python+NumPy programs.

Installation

pip install jax

Documentation

© 2026 AI/ML Engineering Textbook

A comprehensive guide to AI/ML engineering, from mathematical foundations to agentic systems.