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
Unified system for large-scale distributed training and inference.
Guidance language for controlling LLM generation with constraints, grammars, and JSON schemas.
Library for easily accessing and processing ML datasets.
JAX neural network library with PyTorch-like module system.
High-level deep learning API supporting JAX, TensorFlow, and PyTorch backends.
Gradient boosting library by Yandex with native categorical feature handling.