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 anaconda::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

May 12, 2026 at 14:48

License

Apache-2.0

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://docs.jax.dev