Jake VanderPlas 0d4bcde7ca Add experimental/sparse_ops & cusparse wrappers in jaxlib ...

PiperOrigin-RevId: 368663407

2021-04-15 10:11: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

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

Factor out wheel-building functions in build_jaxlib_wheels.sh.

2021-04-09 01:30:36 +00: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