Aart Bik c1261ccd27 Adds a wrapper to sparse tensor dialect, as part of an ...

an initial prototype of an alternate JAX compilation path
that emits the MLIR MHLO/CHLO dialects instead of classic XLA HLO
together with sparse tensor types.

PiperOrigin-RevId: 443438043

2022-04-21 11:48:44 -07:00

..

rocm

Upgrade ROCm build docker to ROCm version 5.1.

2022-03-31 11:56:38 +00:00

build_jaxlib_wheels_helpers.sh

Remove -it flag from docker run command in jaxlib build script.

2021-04-14 09:44:42 -07:00

build_jaxlib_wheels_macos.sh

Drop support for NumPy 1.18.

2022-01-04 12:11:38 -08:00

build_jaxlib_wheels.sh

Drop Python 3.6 support.

2021-07-15 14:20:29 -04:00

build_wheel_docker_entrypoint.sh

Allow manylinux_2_12_x86_64 in auditwheel test. This is an alias by PEP600.

2021-05-07 18:22:28 -04:00

build_wheel.py

Adds a wrapper to sparse tensor dialect, as part of an

2022-04-21 11:48:44 -07:00

BUILD.bazel

Adds a wrapper to sparse tensor dialect, as part of an

2022-04-21 11:48:44 -07:00

build.py

Upgrade the bazel version to 5.1.1

2022-04-12 17:48:09 -07:00

Dockerfile

Drop support for NumPy 1.18.

2022-01-04 12:11:38 -08:00

install_cuda.sh

Update cuDNN versions to highest available.

2021-03-09 19:07:00 +00:00

LICENSE.txt

Refactor JAX CPU kernels to make them usable from C++.

2021-09-03 10:03:54 -07:00

test-requirements.txt

CI: pin pillow<9.1 to prevent deprecation warnings

2022-04-01 09:23:27 -07:00