https://github.com/google/jax
History
Tip revision: f3d07a3222e6bbce5d5c393e8b37e83d2899ff80 authored by Adam Paszke on 05 August 2024, 11:22:32 UTC
[Mosaic TPU] Allow simultaneous column and row shifts
Tip revision: f3d07a3
File Mode Size
__init__.py -rw-r--r-- 870 bytes

back to top