https://github.com/google/jax
History
Tip revision: 720611aea1233d4b5fe0ee07e1e0cc8cbadde277 authored by Dimitar (Mitko) Asenov on 23 November 2023, 15:29:05 UTC
[XLA] Use `mhlo.topk` from JAX instead of `chlo.topk`.
Tip revision: 720611a
File Mode Size
__init__.py -rw-r--r-- 870 bytes

back to top