JAX
High-performance numerical computing library by Google with auto-differentiation.
About
JAX by Google Research is a Python library for accelerator-oriented array computation that pairs a NumPy-like API with automatic differentiation and XLA compilation. It differentiates through native Python and NumPy code including loops and branches, composes forward and reverse-mode autodiff, and offers transforms like jit, vmap, and pmap for compilation and parallelism on GPU and TPU. Released under the Apache 2.0 license.
Reviews (0)
Leave a Review
No reviews yet. Be the first to review!
Details
- Category
- AI Frameworks & Libraries
- Price
- Free
- Platform
- Local/Desktop
- Difficulty
- Advanced (4/5)
- License
- Apache-2.0
- Added
- Apr 3, 2026
Related Tools
Tensor library for machine learning on commodity hardware
Structured output extraction from LLMs with Pydantic
Deploy LangChain runnables as REST APIs
Unified system for large-scale distributed training and inference.
High-level deep learning library making neural nets accessible with best practices.
Open-source machine learning framework by Meta with dynamic computation graphs.