|
..
|
|
cpu
|
|
cuda
|
|
gpu
|
|
mlir
|
|
rocm
|
|
BUILD
|
|
README.md
|
|
cpu_feature_guard.c
|
|
ducc_fft.py
|
|
gpu_linalg.py
|
|
gpu_prng.py
|
|
gpu_rnn.py
|
|
gpu_solver.py
|
|
gpu_sparse.py
|
|
handle_pool.h
|
|
hlo_helpers.py
|
|
init.py
|
|
jax.bzl
|
|
kernel_helpers.h
|
|
kernel_pybind11_helpers.h
|
|
lapack.py
|
|
setup.cfg
|
|
setup.py
|
|
symlink_files.bzl
|