https://github.com/google/jax
History
Tip revision: 8fb4e08f57e063c9f5ec2cc3ee1ebc405fbd5d01 authored by Jevin Jiang on 28 March 2024, 05:11:38 UTC
[XLA:Mosaic] Add type-convert-insertion pass and support shapecast a vector whose dtype is packed and sublane dim size is 1.
Tip revision: 8fb4e08
File Mode Size
lorentz.png -rw-r--r-- 560.6 KB
nested_pmap.png -rw-r--r-- 23.3 KB
wave_movie.gif -rw-r--r-- 5.6 MB

back to top