https://github.com/google/jax
History
Tip revision: 21fc6e0229e0f5f1cb5f1f69d2c3daa2e5c2ca11 authored by jax authors on 11 May 2023, 22:20:30 UTC
Merge pull request #15978 from skye:version
Tip revision: 21fc6e0
File Mode Size
cpu
cuda
gpu
mlir
rocm
BUILD -rw-r--r-- 4.3 KB
README.md -rw-r--r-- 402 bytes
cpu_feature_guard.c -rw-r--r-- 5.2 KB
ducc_fft.py -rw-r--r-- 4.5 KB
gpu_linalg.py -rw-r--r-- 2.4 KB
gpu_prng.py -rw-r--r-- 3.3 KB
gpu_rnn.py -rw-r--r-- 5.0 KB
gpu_solver.py -rw-r--r-- 17.6 KB
gpu_sparse.py -rw-r--r-- 12.2 KB
handle_pool.h -rw-r--r-- 3.3 KB
hlo_helpers.py -rw-r--r-- 4.1 KB
init.py -rw-r--r-- 629 bytes
jax.bzl -rw-r--r-- 8.0 KB
kernel_helpers.h -rw-r--r-- 1.7 KB
kernel_pybind11_helpers.h -rw-r--r-- 1.6 KB
lapack.py -rw-r--r-- 23.9 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
utils.cc -rw-r--r-- 7.3 KB

README.md

back to top