| File | Mode | Size |
|---|---|---|
| cpu | ||
| cuda | ||
| gpu | ||
| mlir | ||
| rocm | ||
| BUILD | -rw-r--r-- | 3.9 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-- | 2.1 KB |
| gpu_rnn.py | -rw-r--r-- | 4.7 KB |
| gpu_solver.py | -rw-r--r-- | 16.5 KB |
| gpu_sparse.py | -rw-r--r-- | 12.2 KB |
| handle_pool.h | -rw-r--r-- | 3.3 KB |
| init.py | -rw-r--r-- | 629 bytes |
| jax.bzl | -rw-r--r-- | 7.6 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.2 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 |
