Revision 423c6c42abd6e9355dd24834e2e100801b21a39b authored by Christos Perivolaropoulos on 02 July 2024, 16:11:27 UTC, committed by jax authors on 02 July 2024, 16:17:26 UTC
PiperOrigin-RevId: 648738847
1 parent 42724eb
File | Mode | Size |
---|---|---|
cpu | ||
cuda | ||
gpu | ||
mlir | ||
mosaic | ||
rocm | ||
tools | ||
triton | ||
BUILD | -rw-r--r-- | 6.7 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.9 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.7 KB |
gpu_prng.py | -rw-r--r-- | 4.6 KB |
gpu_rnn.py | -rw-r--r-- | 5.4 KB |
gpu_solver.py | -rw-r--r-- | 19.0 KB |
gpu_sparse.py | -rw-r--r-- | 13.1 KB |
gpu_triton.py | -rw-r--r-- | 2.3 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.0 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 |
rocm_plugin_extension.cc | -rw-r--r-- | 5.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 |
Computing file changes ...