mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib
History
Mehdi Amini 5a6cb438e8 Move MHLO to XLA 
As part of the OpenXLA project, we're splitting XLA outside of TensorFlow.
MHLO belongs to OpenXLA and we're relocating it nested under XLA to allow the
split. Some further directory layout change will likely happen over time.

PiperOrigin-RevId: 464126676
2022-07-29 11:54:51 -07:00
..
cuda
Replace references to absl::string_view with std::string_view.
2022-05-24 14:21:32 -07:00
mlir
Move MHLO to XLA
2022-07-29 11:54:51 -07:00
rocm
[ROCm] Fixes for compilation failures caused by compiler changes in ROCm Tensorflow fork.
2022-06-29 14:34:08 +00:00
BUILD
Integrate LLVM at llvm/llvm-project@8aff88fd3a
2022-07-19 08:31:24 -07:00
cpu_feature_guard.c
move #endif so that Windows doesn't have GetXCR0EAX defined twice
2021-06-04 17:26:27 -07:00
cpu_kernels.cc
schur lapack wrapper
2021-09-29 14:29:52 +02:00
gpu_linalg.py
[ROCm] Fix hip_lu_pivots_to_permutation.
2022-05-23 13:58:59 +00:00
gpu_prng.py
Fix unit test failures on ROCm.
2022-05-09 13:36:48 +00:00
gpu_solver.py
Add a batched QR decomposition implementation on GPU.
2022-05-18 14:50:18 -07:00
gpu_sparse.py
[sparse] Update _validate_coo_mhlo in gpu_sparse.
2022-07-14 20:35:09 -07:00
handle_pool.h
[JAX] Refactor JAX custom kernels to split kernel implementations from Python bindings.
2021-09-02 07:53:09 -07:00
init.py
CI: fix flake8 ignore declarations
2022-04-21 13:44:12 -07:00
jax.bzl
Delete sharded_jit
2022-07-29 08:19:52 -07:00
kernel_helpers.h
Use the new "custom call status" facility to report errors in jaxlib
2021-08-09 15:06:39 -07:00
kernel_pybind11_helpers.h
[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.
2020-10-23 14:20:32 -07:00
lapack_kernels_using_lapack.cc
Make lapack symbols strong in lapack_kernels
2022-01-25 16:30:30 -08:00
lapack_kernels.cc
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
lapack_kernels.h
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
lapack.cc
Merge pull request #8033 from SaturdayGenfo:schur-lapack-wrapper
2021-09-29 09:37:49 -07:00
lapack.py
Update minimum jaxlib version to 0.3.14.
2022-07-08 00:36:02 +00:00
mhlo_helpers.py
Refactor custom call building code in jaxlib to use a helper function.
2022-05-06 14:51:24 -07:00
pocketfft_kernels.cc
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
pocketfft_kernels.h
Migrate 'jaxlib' CPU custom-calls to the status-returning API
2022-03-29 17:14:14 -07:00
pocketfft.cc
Reapply: Drop flatbuffers as a Python dependency of JAX.
2022-06-28 14:25:14 -07:00
pocketfft.fbs
[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.
2020-10-23 14:20:32 -07:00
pocketfft.py
Update minimum jaxlib version to 0.3.14.
2022-07-08 00:36:02 +00:00
README.md
Populate long_description for jax & jaxlib
2022-07-13 14:03:32 -07:00
setup.cfg
Export LICENSE.txt file in jaxlib wheels.
2020-12-11 10:20:03 -05:00
setup.py
_mlirTransforms merged into _mlirRegisterEverything.
2022-07-20 14:43:27 -07:00
symlink_files.bzl
Use symlink_files() to add version.py to jaxlib, rather than copying it in as part of the wheel assembly process.
2022-07-01 09:07:03 -07:00

README.md

jaxlib: support library for JAX

jaxlib is the support library for JAX. While JAX itself is a pure Python package, jaxlib contains the binary (C/C++) parts of the library, including Python bindings, the XLA compiler, the PJRT runtime, and a handful of handwritten kernels. For more information, including installation and build instructions, refer to main JAX README: https://github.com/google/jax/.