mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
8 lines
402 B
Markdown
8 lines
402 B
Markdown
# 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/jax-ml/jax/.
|