Peter Hawkins 79a7f7bca8 Don't build CUDA 11.2 wheels. ...

Update XLA.

CUDA 11.1 wheels are compatible with CUDA versions 11.1+, since NVidia now promises enhanced version compatibility between CUDA minor releases starting with CUDA 11.1

2021-04-26 09:43:29 -04: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

Change jaxlib build rules to build a wheel, rather than writing output to the source directory.

2020-11-20 11:47:00 -05:00

build_jaxlib_wheels.sh

Don't build CUDA 11.2 wheels.

2021-04-26 09:43:29 -04:00

build_wheel_docker_entrypoint.sh

Don't build with Cloud TPU support for GPU or Mac wheels.

2021-03-08 16:19:19 -08:00

build_wheel.py

Add experimental/sparse_ops & cusparse wrappers in jaxlib

2021-04-15 10:11:10 -07:00

BUILD.bazel

Add experimental/sparse_ops & cusparse wrappers in jaxlib

2021-04-15 10:11:10 -07:00

build.py

Don't build with Cloud TPU support for GPU or Mac wheels.

2021-03-08 16:19:19 -08:00

Dockerfile

Change jaxlib build rules to build a wheel, rather than writing output to the source directory.

2020-11-20 11:47:00 -05:00

generate_release_index.py

Use local version identifiers to distribute cuda jaxlib wheels.

2020-10-09 13:47:54 -07:00

install_cuda.sh

Update cuDNN versions to highest available.

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

LICENSE.txt

Include a LICENSE.txt in jaxlib wheels.

2020-12-11 06:45:15 -08:00

test-requirements.txt

Specify minimum jaxlib version in a single location

2021-03-22 16:14:41 -07:00