Yash Katariya ac0796048f Move cuda .py files to :gpu_support so that if :gpu_support is not present, then internal jaxlib will act like a CPU jaxlib even if --config=cuda is specified. ...

PiperOrigin-RevId: 403170945

2021-10-14 13:20:01 -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

Refactor JAX CPU kernels to make them usable from C++.

2021-09-03 10:03:54 -07: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] Fix pylint errors.

2021-10-04 17:54:46 -07:00

BUILD.bazel

Move cuda .py files to :gpu_support so that if :gpu_support is not present, then internal jaxlib will act like a CPU jaxlib even if --config=cuda is specified.

2021-10-14 13:20:01 -07:00

build.py

[JAX] Fix pylint errors.

2021-10-04 17:54:46 -07:00

Dockerfile

Refactor JAX CPU kernels to make them usable from C++.

2021-09-03 10:03:54 -07:00

install_cuda.sh

Update cuDNN versions to highest available.

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

LICENSE.txt

Refactor JAX CPU kernels to make them usable from C++.

2021-09-03 10:03:54 -07:00

test-requirements.txt

Adds a Wadler-Lindig pretty printer.

2021-09-27 21:09:24 -04:00