Skip to content

Latest commit

 

History

History
 
 

jaxlib

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
cuda
cuda
 
 
mlir
mlir
 
 
rocm
rocm
 
 
BUILD
BUILD
 
 
README.md
README.md
 
 
cpu_feature_guard.c
cpu_feature_guard.c
 
 
cpu_kernels.cc
cpu_kernels.cc
 
 
gpu_linalg.py
gpu_linalg.py
 
 
gpu_prng.py
gpu_prng.py
 
 
gpu_solver.py
gpu_solver.py
 
 
gpu_sparse.py
gpu_sparse.py
 
 
handle_pool.h
handle_pool.h
 
 
init.py
init.py
 
 
jax.bzl
jax.bzl
 
 
kernel_helpers.h
kernel_helpers.h
 
 
kernel_pybind11_helpers.h
kernel_pybind11_helpers.h
 
 
lapack.cc
lapack.cc
 
 
lapack.py
lapack.py
 
 
lapack_kernels.cc
lapack_kernels.cc
 
 
lapack_kernels.h
lapack_kernels.h
 
 
lapack_kernels_using_lapack.cc
lapack_kernels_using_lapack.cc
 
 
mhlo_helpers.py
mhlo_helpers.py
 
 
pocketfft.cc
pocketfft.cc
 
 
pocketfft.fbs
pocketfft.fbs
 
 
pocketfft.py
pocketfft.py
 
 
pocketfft_kernels.cc
pocketfft_kernels.cc
 
 
pocketfft_kernels.h
pocketfft_kernels.h
 
 
setup.cfg
setup.cfg
 
 
setup.py
setup.py
 
 
symlink_files.bzl
symlink_files.bzl
 
 

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