JAX

High-performance numerical computing library by Google with auto-differentiation.

Open SourceSelf HostedOffline Capable
0.0 (0)

About

JAX by Google Research combines NumPy with automatic differentiation and XLA compilation for high-performance ML research. Supports GPU/TPU acceleration, vmap, pmap for parallelism, and JIT compilation. Apache 2.0 license.

Reviews (0)

Leave a Review

No reviews yet. Be the first to review!

Details

Price
Free
Platform
Local/Desktop
Difficulty
Advanced (4/5)
License
Apache-2.0
Added
Apr 3, 2026

Similar Tools

Featured

Open-source machine learning framework by Meta with dynamic computation graphs.

Open SourceSelf HostedOffline
Intermediate
0.0 (0)
Featured

End-to-end open-source ML platform by Google for training and deployment.

Open SourceSelf HostedOffline
Intermediate
0.0 (0)

High-level deep learning API supporting JAX, TensorFlow, and PyTorch backends.

Open SourceSelf HostedOffline
Easy
0.0 (0)