Skip to content

Latest commit

 

History

History

jaxlib

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
cpu
cpu
 
 
cuda
cuda
 
 
gpu
gpu
 
 
mlir
mlir
 
 
mosaic
mosaic
 
 
rocm
rocm
 
 
tools
tools
 
 
triton
triton
 
 
BUILD
BUILD
 
 
README.md
README.md
 
 
absl_status_casters.h
absl_status_casters.h
 
 
cpu_feature_guard.c
cpu_feature_guard.c
 
 
cuda_plugin_extension.cc
cuda_plugin_extension.cc
 
 
ffi_helpers.h
ffi_helpers.h
 
 
gpu_common_utils.py
gpu_common_utils.py
 
 
gpu_linalg.py
gpu_linalg.py
 
 
gpu_prng.py
gpu_prng.py
 
 
gpu_rnn.py
gpu_rnn.py
 
 
gpu_solver.py
gpu_solver.py
 
 
gpu_sparse.py
gpu_sparse.py
 
 
gpu_triton.py
gpu_triton.py
 
 
handle_pool.h
handle_pool.h
 
 
hlo_helpers.py
hlo_helpers.py
 
 
init.py
init.py
 
 
jax.bzl
jax.bzl
 
 
kernel_helpers.h
kernel_helpers.h
 
 
kernel_nanobind_helpers.h
kernel_nanobind_helpers.h
 
 
lapack.py
lapack.py
 
 
rocm_plugin_extension.cc
rocm_plugin_extension.cc
 
 
setup.py
setup.py
 
 
symlink_files.bzl
symlink_files.bzl
 
 
utils.cc
utils.cc
 
 

README.md

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/.