mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 20:36:05 +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/google/jax/.
|