jaxlib

Anaconda Verified

Composable transformations of Python+NumPy programs - differentiate, vectorize, JIT to GPU/TPU, and more

Versions

Installation

To install this package, run one of the following:

Conda

$conda install main::jaxlib

Usage Tracking

Version

5 / 8 versions selected

Downloads (Last 6 months): 0

Description

JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning.

About

Summary

Composable transformations of Python+NumPy programs - differentiate, vectorize, JIT to GPU/TPU, and more

Last Updated

Dec 18, 2025 at 00:14

License

Apache-2.0

Total Downloads

2.4K

Supported Platforms

macOS-arm64

linux-64

linux-aarch64

win-64

Unsupported Platforms

macOS-64 Last supported version: 0.4.35

Home

https://github.com/jax-ml/jax

GitHub Repository

https://github.com/jax-ml/jax

Documentation

https://github.com/jax-ml/jax/blob/main/README.md