**Breaking changes:**
* `SaliencyParams` has moved from `audiotree.datasources.SaliencyParams` to `audiotree.SaliencyParams`
Updates:
The code has been tested with device parallel sharding. See the recent updates to [DAC-JAX](https://github.com/DBraun/DAC-JAX/blob/main/scripts/input_pipeline.py).
SaliencyParams has a new `search_function` parameter. The two valid strings are `SaliencyParams.search_uniform` and `SaliencyParams.search_early_bias`. You can also plug in your own Callable function.