Yash Katariya 6506ee2a40 Copybara import of the project: ...

--
57af5360a1ca1356dbf7760c76e241f7134ef6dd by Jake VanderPlas <jakevdp@google.com>:

[Rollback] Update required Python version to 3.9

PiperOrigin-RevId: 528905991

2023-05-02 15:33:29 -07:00

..

rocm

[ROCm]: Run pmap test on specific number of GPUs

2023-03-30 18:34:47 +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

[JAX] Improve handling of metadata in compilation cache.

2023-04-19 13:27:04 -07:00

BUILD.bazel

Delete remote TPU support.

2023-03-24 12:33:33 -07:00

build.py

Copybara import of the project:

2023-05-02 15:33:29 -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

Switch from pocketfft to ducc

2022-08-26 13:36:25 +00:00

parallel_accelerator_execute.sh

Add script for parallel accelerator testing under Bazel.

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

test-requirements.txt

Remove pytest_benchmark for test-requirements.txt and move the benchmark file which was using that package to use google_benchmark.

2022-10-25 11:59:32 -07:00