swh:1:snp:71afc25eb6e6e055a37a962e6b91010ec35e397f
History
Tip revision: f4158ace933482844c145a6b919bf5dc86e084ba authored by jax authors on 18 June 2024, 12:21:34 UTC
Merge pull request #21949 from hawkinsp:winwheel
Tip revision: f4158ac
File Mode Size
cpu
cuda
gpu
mlir
mosaic
rocm
tools
triton
BUILD -rw-r--r-- 6.0 KB
README.md -rw-r--r-- 402 bytes
absl_status_casters.h -rw-r--r-- 7.7 KB
cpu_feature_guard.c -rw-r--r-- 5.6 KB
cuda_plugin_extension.cc -rw-r--r-- 4.8 KB
ducc_fft.py -rw-r--r-- 2.9 KB
gpu_common_utils.py -rw-r--r-- 905 bytes
gpu_linalg.py -rw-r--r-- 3.8 KB
gpu_prng.py -rw-r--r-- 3.5 KB
gpu_rnn.py -rw-r--r-- 5.4 KB
gpu_solver.py -rw-r--r-- 18.7 KB
gpu_sparse.py -rw-r--r-- 13.0 KB
gpu_triton.py -rw-r--r-- 2.1 KB
handle_pool.h -rw-r--r-- 3.3 KB
hlo_helpers.py -rw-r--r-- 10.1 KB
init.py -rw-r--r-- 629 bytes
jax.bzl -rw-r--r-- 10.2 KB
kernel_helpers.h -rw-r--r-- 1.7 KB
kernel_nanobind_helpers.h -rw-r--r-- 2.3 KB
lapack.py -rw-r--r-- 24.2 KB
setup.py -rw-r--r-- 3.4 KB
symlink_files.bzl -rw-r--r-- 6.0 KB
utils.cc -rw-r--r-- 7.2 KB

README.md

back to top