https://github.com/google/jax
History
Tip revision: 161b1b1c7e5a964bd6335c2350951aca59d20538 authored by Adam Paszke on 05 June 2024, 15:19:49 UTC
[Mosaic TPU] Add support for semaphore operands (inputs and outputs)
Tip revision: 161b1b1
File Mode Size
build_wheels
tools
Dockerfile.ms -rw-r--r-- 2.5 KB
README.md -rw-r--r-- 2.6 KB
build_common.sh -rw-r--r-- 1.7 KB
build_rocm.sh -rwxr-xr-x 1.0 KB
ci_build -rwxr-xr-x 7.8 KB
ci_build.sh -rwxr-xr-x 4.7 KB
custom_install.sh -rw-r--r-- 996 bytes
run_multi_gpu.sh -rwxr-xr-x 1.6 KB
run_single_gpu.py -rwxr-xr-x 6.6 KB
setup.rocm.sh -rwxr-xr-x 2.9 KB

README.md

back to top