Cloud Han 2d321c26e6 Use TF_CUDA_PATHS ...

CUDA_TOOLKIT_PATH and CUDNN_INSTALL_PATH are deprecated, see TF 2.0
release notes for more information

2021-07-18 22:55:34 +08: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 Python 3.6 support.

2021-07-15 14:20:29 -04: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

fix machine tag, on windows platforms.machine() returns AMD64 instread of x64_64

2021-07-18 22:55:02 +08:00

BUILD.bazel

Add a CPU feature guard module to JAX.

2021-05-19 10:58:35 -07:00

build.py

Use TF_CUDA_PATHS

2021-07-18 22:55:34 +08:00

Dockerfile

Bump np versions for build script.

2021-06-22 16:29:56 -07:00

generate_release_indexes.py

Update generate_release_index[es].py to also produce libtpu_releases.html.

2021-07-12 16:02:54 -07:00

install_cuda.sh

Update cuDNN versions to highest available.

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

LICENSE.txt

Rename master to main in embedded links.

2021-06-18 10:00:01 +03:00

test-requirements.txt

Update minimum jaxlib version to 0.1.69.

2021-07-15 17:00:13 -04:00