https://github.com/google/jax
History
Tip revision: 58d516c49ee75bbf5f5ba4731ea2a1261329e4ce authored by jax authors on 12 October 2022, 15:23:43 UTC
Merge pull request #12764 from midjourney:gda_sharding_property
Tip revision: 58d516c
File Mode Size
cuda
mlir
rocm
BUILD -rw-r--r-- 5.5 KB
README.md -rw-r--r-- 402 bytes
cpu_feature_guard.c -rw-r--r-- 5.2 KB
cpu_kernels.cc -rw-r--r-- 6.0 KB
ducc_fft.cc -rw-r--r-- 2.4 KB
ducc_fft.fbs -rw-r--r-- 1003 bytes
ducc_fft.py -rw-r--r-- 4.5 KB
ducc_fft_kernels.cc -rw-r--r-- 4.3 KB
ducc_fft_kernels.h -rw-r--r-- 803 bytes
gpu_linalg.py -rw-r--r-- 2.3 KB
gpu_prng.py -rw-r--r-- 2.1 KB
gpu_solver.py -rw-r--r-- 16.3 KB
gpu_sparse.py -rw-r--r-- 12.1 KB
handle_pool.h -rw-r--r-- 3.3 KB
init.py -rw-r--r-- 629 bytes
jax.bzl -rw-r--r-- 7.1 KB
kernel_helpers.h -rw-r--r-- 1.7 KB
kernel_pybind11_helpers.h -rw-r--r-- 1.7 KB
lapack.cc -rw-r--r-- 9.6 KB
lapack.py -rw-r--r-- 19.5 KB
lapack_kernels.cc -rw-r--r-- 25.6 KB
lapack_kernels.h -rw-r--r-- 6.6 KB
lapack_kernels_using_lapack.cc -rw-r--r-- 3.9 KB
mhlo_helpers.py -rw-r--r-- 2.8 KB
setup.cfg -rw-r--r-- 38 bytes
setup.py -rw-r--r-- 2.4 KB
symlink_files.bzl -rw-r--r-- 6.0 KB

README.md

back to top