CMD + K

jaxlib

Community

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

Installation

To install this package, run one of the following:

Conda
$conda install anaconda-python-gte-3.10::jaxlib

Usage Tracking

0.4.35
0.4.25
0.4.23
0.4.16
0.3.25
5 / 8 versions selected
Downloads (Last 6 months): 0

Description

jaxlib is the support library for JAX. While JAX itself is a pure Python package, jaxlib contains the binary (C/C++) parts of the library, including Python bindings, the XLA compiler, the PJRT runtime, and a handful of handwritten kernels.

About

Summary

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

Last Updated

Mar 31, 2025 at 22:02

License

Apache-2.0

Total Downloads

0

Version Downloads

0

Supported Platforms

linux-64