|
..
|
|
cpu
|
|
cuda
|
|
gpu
|
|
mlir
|
|
mosaic
|
|
rocm
|
|
tools
|
|
triton
|
|
xla
|
|
BUILD
|
|
README.md
|
|
absl_status_casters.h
|
|
cpu_feature_guard.c
|
|
ffi_helpers.h
|
|
gpu_common_utils.py
|
|
gpu_linalg.py
|
|
gpu_prng.py
|
|
gpu_rnn.py
|
|
gpu_solver.py
|
|
gpu_sparse.py
|
|
gpu_triton.py
|
|
hlo_helpers.py
|
|
init.py
|
|
jax.bzl
|
|
jax_common.json
|
|
jax_python_wheel.bzl
|
|
kernel_helpers.h
|
|
kernel_nanobind_helpers.h
|
|
lapack.py
|
|
libjax_common.lds
|
|
libjax_common_darwin.lds
|
|
plugin_support.py
|
|
pyinit_stub.c
|
|
pywrap.bzl
|
|
setup.py
|
|
symlink_files.bzl
|
|
utils.cc
|
|
weakref_lru_cache.cc
|
|
weakref_lru_cache.pyi
|
|
weakref_lru_cache_test.py
|
|
xla_client.py
|
|
xla_extension.py
|