Adam Paszke 8e3f5b1018 Initial commit for Mosaic GPU ...

Moving this to JAX to make it easier to explore Pallas integration.

PiperOrigin-RevId: 625982382

2024-04-18 04:04:10 -07:00

..

rocm

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

2024-01-24 15:08:35 +00:00

build.py

Initial commit for Mosaic GPU

2024-04-18 04:04:10 -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

test-requirements.txt

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

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