Home|About|Gaming cabinet|Archives|How it works|Starter sites

GeezerVison is showing github.com — jax/jaxlib at main · jax-ml/jax

Expand file tree

/

jaxlib

/

Copy path

Directory actions

More options

More options

Directory actions

More options

More options

Latest commit

History

History

/

jaxlib

/

Copy path

Folders and files

Name		Name	Last commit message	Last commit date
parent directory ..
_jax		_jax
cpu		cpu
cuda		cuda
gpu		gpu
mlir		mlir
mosaic		mosaic
oneapi		oneapi
rocm		rocm
thunky		thunky
tools		tools
triton		triton
BUILD		BUILD
README.md		README.md
_ifrt_proxy.pyi		_ifrt_proxy.pyi
_pathways.pyi		_pathways.pyi
_pretty_printer.cc		_pretty_printer.cc
_pretty_printer.pyi		_pretty_printer.pyi
_sdy_mpmd.pyi		_sdy_mpmd.pyi
_xla.pyi		_xla.pyi
absl_status_casters.h		absl_status_casters.h
backport_stubs.py		backport_stubs.py
cached_py_object.h		cached_py_object.h
call_location.cc		call_location.cc
call_location.h		call_location.h
callback.cc		callback.cc
callback.h		callback.h
config.cc		config.cc
config.h		config.h
config_test.py		config_test.py
cpu_feature_guard.c		cpu_feature_guard.c
cpu_feature_guard.pyi		cpu_feature_guard.pyi
cpu_sparse.py		cpu_sparse.py
custom_call_sharding.cc		custom_call_sharding.cc
custom_call_sharding.h		custom_call_sharding.h
digamma_coeffs.h		digamma_coeffs.h
dlpack.cc		dlpack.cc
dlpack.h		dlpack.h
exception_test.py		exception_test.py
ffi.cc		ffi.cc
ffi.h		ffi.h
ffi_helpers.h		ffi_helpers.h
free_threading.h		free_threading.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
guard_lib.cc		guard_lib.cc
guard_lib.h		guard_lib.h
hash_util.h		hash_util.h
ifrt_proxy.cc		ifrt_proxy.cc
init.py		init.py
jax.bzl		jax.bzl
jax.cc		jax.cc
jax_common.json		jax_common.json
jax_jit.cc		jax_jit.cc
jax_jit.h		jax_jit.h
jax_python_wheel.bzl		jax_python_wheel.bzl
kernel_nanobind_helpers.h		kernel_nanobind_helpers.h
lapack.py		lapack.py
lgamma_coeffs.h		lgamma_coeffs.h
libjax_common.lds		libjax_common.lds
libjax_common_darwin.lds		libjax_common_darwin.lds
log_coeffs.h		log_coeffs.h
mlir.cc		mlir.cc
mlir.h		mlir.h
nb_class_ptr.h		nb_class_ptr.h
numpy.cc		numpy.cc
numpy.h		numpy.h
partition_spec.cc		partition_spec.cc
partition_spec.h		partition_spec.h
pathways.cc		pathways.cc
pjit.cc		pjit.cc
pjit.h		pjit.h
plugin_support.py		plugin_support.py
pprof_profile_builder.cc		pprof_profile_builder.cc
pprof_profile_builder.h		pprof_profile_builder.h
py_array.cc		py_array.cc
py_array.h		py_array.h
py_client.cc		py_client.cc
py_client.h		py_client.h
py_client_cpu.cc		py_client_cpu.cc
py_client_cpu.h		py_client_cpu.h
py_compile_only_client.cc		py_compile_only_client.cc
py_compile_only_client.h		py_compile_only_client.h
py_device.cc		py_device.cc
py_device.h		py_device.h
py_device_list.cc		py_device_list.cc
py_device_list.h		py_device_list.h
py_executable.cc		py_executable.cc
py_executable.h		py_executable.h
py_host_callback.cc		py_host_callback.cc
py_host_callback.h		py_host_callback.h
py_host_callback.proto		py_host_callback.proto
py_memory_space.cc		py_memory_space.cc
py_memory_space.h		py_memory_space.h
py_mpmd_loaded_executable.cc		py_mpmd_loaded_executable.cc
py_mpmd_loaded_executable.h		py_mpmd_loaded_executable.h

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