High level operations
Examples
Settings
Lower level operations
Pre- and postprocessing
dict_zeros_like()
Source code
Wrapper around zeros_like, that works also for dicts with jnp.ndarray entries.