mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib
History
Eugene Burmako f337c00ed5 Remove *_mhlo compatibility shims from jaxlib 
We introduced these shims when migrating from MHLO to StableHLO, and they helped accommodate the version skew between jaxlib and JAX across different environments. Now that a sufficient amount of time has passed, these shims are no longer used anywhere and can be deleted.

PiperOrigin-RevId: 510820007
2023-02-19 09:03:14 -08:00
..
cpu
Remove an unnecessary include/ from pybind11 include paths.
2022-11-30 14:20:02 -08:00
cuda
Run GPU presubmits via bazel test on the RBE cluster. This speeds up the build + testing significantly (upto 10x).
2022-11-29 08:45:58 -08:00
gpu
Add support for dynamic shapes to GPU threefry2x32 custom call.
2022-12-27 04:48:26 -08:00
mlir
jaxlib: fix mlir_hlo build rule
2022-11-16 15:42:05 -08:00
rocm
Merge CUDA and ROCM kernel code in jaxlib.
2022-10-25 07:23:34 -07:00
BUILD
(NFC) Prepare for migration from producing MHLO to producing StableHLO
2022-12-15 21:00:07 -08:00
cpu_feature_guard.c
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
ducc_fft.py
Remove *_mhlo compatibility shims from jaxlib
2023-02-19 09:03:14 -08:00
gpu_linalg.py
(NFC) Prepare for migration from producing MHLO to producing StableHLO
2022-12-15 21:00:07 -08:00
gpu_prng.py
Add support for dynamic shapes to GPU threefry2x32 custom call.
2022-12-27 04:48:26 -08:00
gpu_rnn.py
Migrate JAX from producing MHLO to producing StableHLO
2022-12-27 08:53:20 -08:00
gpu_solver.py
Bump minimum jaxlib_version to 0.4.2 i.e xla_extension_version == 119 and mlir_api_version == 43
2023-02-06 10:37:33 -08:00
gpu_sparse.py
fix change to csr lowering rule
2023-02-13 08:39:05 -08:00
handle_pool.h
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hlo_helpers.py
Migrate JAX from producing MHLO to producing StableHLO
2022-12-27 08:53:20 -08:00
init.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
jax.bzl
Limit visibility of Bazel target jax:global_device_array.
2023-02-17 14:30:05 -08:00
kernel_helpers.h
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
kernel_pybind11_helpers.h
Remove an unnecessary include/ from pybind11 include paths.
2022-11-30 14:20:02 -08:00
lapack.py
Remove *_mhlo compatibility shims from jaxlib
2023-02-19 09:03:14 -08: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
Drop support for Python 3.7
2022-11-29 15:01:47 -08:00
symlink_files.bzl
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -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/.