Skip to main content

XLA library for JAX

Project description

jaxlib: support library for JAX

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. For more information, including installation and build instructions, refer to main JAX README: https://github.com/google/jax/.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

jaxlib_rocm-0.4.6.540-cp310-cp310-manylinux2014_x86_64.whl (90.7 MB view details)

Uploaded CPython 3.10

jaxlib_rocm-0.4.6.540-cp39-cp39-manylinux2014_x86_64.whl (90.7 MB view details)

Uploaded CPython 3.9

jaxlib_rocm-0.4.6.540-cp38-cp38-manylinux2014_x86_64.whl (90.7 MB view details)

Uploaded CPython 3.8

File details

Details for the file jaxlib_rocm-0.4.6.540-cp310-cp310-manylinux2014_x86_64.whl.

File metadata

File hashes

Hashes for jaxlib_rocm-0.4.6.540-cp310-cp310-manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 e71d2f26102e985803298abc2a2349a9d1eeaf9c987e78802c6bcbd3e6bdcb76
MD5 c0b07d6fca411af0ac6f1260fdc9f03a
BLAKE2b-256 392871606bb0063b93b3d3f1b50e6c8bd4cd66a2a91d62e7b0c40afdb554d421

See more details on using hashes here.

File details

Details for the file jaxlib_rocm-0.4.6.540-cp39-cp39-manylinux2014_x86_64.whl.

File metadata

File hashes

Hashes for jaxlib_rocm-0.4.6.540-cp39-cp39-manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 19378f021be08f4229129a84d135171edaced23962b43230634e08f05e0361d4
MD5 b2f8ded736694cefcd7fe3bd58b0f74b
BLAKE2b-256 f1e0474eed284094d9cb67cd1b9efa86362bfbd6ac265af486d4473f3ef828b6

See more details on using hashes here.

File details

Details for the file jaxlib_rocm-0.4.6.540-cp38-cp38-manylinux2014_x86_64.whl.

File metadata

File hashes

Hashes for jaxlib_rocm-0.4.6.540-cp38-cp38-manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 eb967ed32517c00673e3c9c12bb090a602280429957d54b8f1cd7667b6457732
MD5 e175a9fcc35fdf9751d780949de7d30f
BLAKE2b-256 330adbff969b8894ca5ea6736d164e229a6d73fa14477c18a0bb02d93c457d6f

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page