Jieying Luo abe7982d65 Remove enable_gpu and xla_python_enable_gpu from jax .bazelrc. ...

The plugin is released and the flag is no longer needed.

Also set default value of enable_gpu to False. enable_gpu will be removed in the next change.

PiperOrigin-RevId: 660059432

2024-08-06 12:39:45 -07:00

..

rocm

Fix lint in run_single_gpu.py

2024-08-02 16:03:03 -07:00

BUILD.bazel

Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details

2024-05-15 18:20:56 -07:00

build.py

Remove enable_gpu and xla_python_enable_gpu from jax .bazelrc.

2024-08-06 12:39:45 -07:00

collect-profile-requirements.txt

Cloud TPU CI: make sure we update test deps and upgrade protobuf version

2023-07-19 16:17:47 -07:00

parallel_accelerator_execute.sh

Add script for parallel accelerator testing under Bazel.

2022-07-06 10:58:04 -04:00

requirements_lock_3_10.txt

Use NumPy 2.0.0 and SciPy 1.13.1 in builds.

2024-06-17 19:35:08 +00:00

requirements_lock_3_11.txt

Use NumPy 2.0.0 and SciPy 1.13.1 in builds.

2024-06-17 19:35:08 +00:00

requirements_lock_3_12.txt

Use NumPy 2.0.0 and SciPy 1.13.1 in builds.

2024-06-17 19:35:08 +00:00

requirements_lock_3_13.txt

Bump scipy from 1.13.0 to 1.13.1

2024-06-17 19:58:11 +00:00

requirements.in

Update minimum Python version to 3.10.

2024-06-26 13:47:14 -04:00

test-requirements.txt

Added filelock to test-requirements.txt and requirements lock files

2024-06-11 11:53:10 +01:00