Peter Hawkins 03876bd702 build.py fixes. ...

* Add aarch64 as a known target_cpu value.
* Only pass --bazel_options to build actions since they can make "bazel
  shutdown" fail.
* Pass the bazel startup options to "bazel shutdown".

Issue https://github.com/google/jax/issues/7097
Fixes https://github.com/google/jax/issues/7639

2022-08-16 15:47:15 +00:00

..

rocm

[ROCm] Update Dockerfile.rocm.

2022-08-09 11:09:10 -07: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

_mlirTransforms merged into _mlirRegisterEverything.

2022-07-20 14:43:27 -07:00

BUILD.bazel

Populate long_description for jax & jaxlib

2022-07-13 14:03:32 -07:00

build.py

build.py fixes.

2022-08-16 15:47:15 +00: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

parallel_accelerator_execute.sh

Add script for parallel accelerator testing under Bazel.

2022-07-06 10:58:04 -04:00

test-requirements.txt

Tests: require pillow>=9.1.0 & remove backward compatibility

2022-08-12 13:34:56 -07:00