Jake VanderPlas f9a4162551 Specify minimum jaxlib version in a single location

2021-03-22 16:14:41 -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

Add CUDA 11.2 to wheel builds.

2021-02-08 09:54:33 -05: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

Move flake8 & mypy checks to pre-commit

2021-02-10 13:44:40 -08:00

BUILD.bazel

Merge pull request #5115 from inailuig:rocm-gpukernels

2020-12-17 14:01:04 -08: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