https://github.com/google/jax
History
Tip revision: 52c41bf9699e6a3c5e609cdb41442226d3bcf9a3 authored by Tomás Longeri on 31 May 2024, 06:07:18 UTC
[Mosaic] In infer-vector-layout's elementwise rule, when there's a conflict, we can still sometimes keep a replicated offset
Tip revision: 52c41bf
File Mode Size
rocm
BUILD.bazel -rw-r--r-- 1.9 KB
build.py -rwxr-xr-x 23.8 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-- 565 bytes
requirements_lock_3_10.txt -rw-r--r-- 44.5 KB
requirements_lock_3_11.txt -rw-r--r-- 44.1 KB
requirements_lock_3_12.txt -rw-r--r-- 44.1 KB
requirements_lock_3_13.txt -rw-r--r-- 2.3 KB
requirements_lock_3_9.txt -rw-r--r-- 44.9 KB
test-requirements.txt -rw-r--r-- 149 bytes

back to top