https://github.com/google/jax
History
Tip revision: 4ed8f5a895e799237ff17b9a6574a15f454a1776 authored by Tom Cobley on 11 July 2024, 17:52:02 UTC
Allow implementations of XLA:CPU collectives to be passed directly into `make_cpu_client`.
Tip revision: 4ed8f5a
File Mode Size
rocm
BUILD.bazel -rw-r--r-- 1.9 KB
build.py -rwxr-xr-x 25.0 KB
collect-profile-requirements.txt -rw-r--r-- 101 bytes
parallel_accelerator_execute.sh -rwxr-xr-x 3.3 KB
requirements.in -rw-r--r-- 324 bytes
requirements_lock_3_10.txt -rw-r--r-- 44.7 KB
requirements_lock_3_11.txt -rw-r--r-- 44.3 KB
requirements_lock_3_12.txt -rw-r--r-- 44.3 KB
requirements_lock_3_13.txt -rw-r--r-- 2.3 KB
test-requirements.txt -rw-r--r-- 158 bytes

back to top