JAX
View on GitHub
35.0k
High-performance machine learning research tool for high-speed numerical computing
About JAX
High-performance machine learning research tool for high-speed numerical computing
Key Features
- Automatic differentiation
- GPU/TPU support
- High performance
- Numpy integration
Installation
uv add jaxRelated Tools
Theano
A Python library to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays
10.0k stars
TensorFlow
An end-to-end open source machine learning platform
194.0k stars
PyTorch
An open source machine learning framework that accelerates research to production
98.0k stars
Keras
A deep learning API written in Python, running on top of TensorFlow
63.9k stars
Quick Info
- Latest Version
- 0.4.10
- First Release
- 2018
- License
- Apache-2.0