https://github.com/google/jax
History
Tip revision: 5b576cb03e85f2e4d8d99f314cf5d87981ef5d42 authored by Peter Hawkins on 27 June 2022, 20:56:00 UTC
Revert: Drop flatbuffers as a Python dependency of JAX.
Tip revision: 5b576cb
File Mode Size
cuda
mlir
rocm
BUILD -rw-r--r-- 4.6 KB
cpu_feature_guard.c -rw-r--r-- 5.2 KB
cpu_kernels.cc -rw-r--r-- 6.0 KB
gpu_linalg.py -rw-r--r-- 2.3 KB
gpu_prng.py -rw-r--r-- 2.1 KB
gpu_solver.py -rw-r--r-- 15.1 KB
gpu_sparse.py -rw-r--r-- 11.3 KB
handle_pool.h -rw-r--r-- 3.3 KB
init.py -rw-r--r-- 623 bytes
jax.bzl -rw-r--r-- 3.7 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-- 20.5 KB
lapack_kernels.cc -rw-r--r-- 25.4 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.4 KB
pocketfft.cc -rw-r--r-- 1.0 KB
pocketfft.fbs -rw-r--r-- 1009 bytes
pocketfft.py -rw-r--r-- 7.7 KB
pocketfft_kernels.cc -rw-r--r-- 3.8 KB
pocketfft_kernels.h -rw-r--r-- 799 bytes
setup.cfg -rw-r--r-- 38 bytes
setup.py -rw-r--r-- 2.1 KB
version.py -rw-r--r-- 925 bytes

back to top