Skip to content
Snippets Groups Projects
Select Git revision
20 results
Compare
user avatar
jit a few internal functions to make them work with global device arrays (#1507)
Clemens Giuliani authored 
i.e. arrays which are `isinstance(x, jax.Array) and not x.is_fully_addressable`.
639aff69
History
user avatar 639aff69
History
Name Last commit Last update