https://github.com/google/jax
History
Tip revision: 7495f71ea42db7b985427a12a1d366ed3cffda64 authored by jax authors on 03 February 2021, 17:15:38 UTC
Merge pull request #5614 from hawkinsp:jaxlib
Tip revision: 7495f71
File Mode Size
BUILD -rw-r--r-- 7.4 KB
cublas.cc -rw-r--r-- 11.1 KB
cuda_gpu_kernel_helpers.cc -rw-r--r-- 1.5 KB
cuda_gpu_kernel_helpers.h -rw-r--r-- 1.3 KB
cuda_prng.py -rw-r--r-- 2.0 KB
cuda_prng_kernels.cc -rw-r--r-- 1.2 KB
cuda_prng_kernels.cu.cc -rw-r--r-- 4.2 KB
cuda_prng_kernels.h -rw-r--r-- 1.0 KB
cusolver.cc -rw-r--r-- 44.9 KB
cusolver.py -rw-r--r-- 14.3 KB
handle_pool.h -rw-r--r-- 2.9 KB
init.py -rw-r--r-- 630 bytes
kernel_helpers.h -rw-r--r-- 1.6 KB
kernel_pybind11_helpers.h -rw-r--r-- 1.7 KB
lapack.pyx -rw-r--r-- 60.2 KB
pocketfft.cc -rw-r--r-- 4.0 KB
pocketfft.fbs -rw-r--r-- 1009 bytes
pocketfft.py -rw-r--r-- 5.1 KB
rocblas.cc -rw-r--r-- 32.0 KB
rocm_gpu_kernel_helpers.cc -rw-r--r-- 1.5 KB
rocm_gpu_kernel_helpers.h -rw-r--r-- 1.3 KB
rocsolver.py -rw-r--r-- 12.9 KB
setup.cfg -rw-r--r-- 38 bytes
setup.py -rw-r--r-- 1.3 KB
version.py -rw-r--r-- 719 bytes

back to top