Adam Paszke cfe64cd5ce [Mosaic GPU] Integrate the ExecutionEngine with the jaxlib GPU plugin ...

This lets us avoid bundling a whole another copy of LLVM with JAX packages
and so we can finally start building Mosaic GPU by default.

PiperOrigin-RevId: 638569750

2024-05-30 01:46:23 -07:00

..

rocm

[ROCM]: Generating pytest html logs from unit-tests.

2024-01-24 15:08:35 +00: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

[Mosaic GPU] Integrate the ExecutionEngine with the jaxlib GPU plugin

2024-05-30 01:46:23 -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_9.txt

fix: Update hermetic python dependencies to numpy=2.0.0rc2 and scipy=1.13.0 for all python version

2024-05-18 23:39:09 -07:00

requirements_lock_3_10.txt

fix: Update hermetic python dependencies to numpy=2.0.0rc2 and scipy=1.13.0 for all python version

2024-05-18 23:39:09 -07:00

requirements_lock_3_11.txt

fix: Update hermetic python dependencies to numpy=2.0.0rc2 and scipy=1.13.0 for all python version

2024-05-18 23:39:09 -07:00

requirements_lock_3_12.txt

fix: Update hermetic python dependencies to numpy=2.0.0rc2 and scipy=1.13.0 for all python version

2024-05-18 23:39:09 -07:00

requirements_lock_3_13.txt

fix: Update hermetic python dependencies to numpy=2.0.0rc2 and scipy=1.13.0 for all python version

2024-05-18 23:39:09 -07:00

requirements.in

Pin matplotlib < 3.9 for Python 3.10 and earlier.

2024-05-22 15:07:09 +00:00

test-requirements.txt

Evaluate the correctness of JAX complex functions using mpmath as a reference

2024-03-21 23:35:29 +02:00