* add interpolate.py

* Add jax dispatch for `searchsorted`

* Import user-facing functions in `tensor.__init__`

* Add for loop based scalar loop

* Pass all loop tests

* Fetch constants from op

* Add while loop test

* Fix while loop and nasty stack over dtypes

* Disable compile here based on CI result

* Fix mypy signature

* Remove unnecessary torch stack

* Only call .cpu when necessary

* Recursive false for torch compiler

* Add elemwise test

* Late import torch

* Do iteration instead of vmap for elemwise

* Clean up and add description

* Add unit test to verify iteration

* Refactor to ravel method

* Fix unpacking
Co-authored-by: Ricardo Vieira <28983449+ricardoV94@users.noreply.github.com>

* Fix comment

* Remove extra return

* Update test

* Add single carry test

* Remove compiler disable

* Better name

* Lint

* Better docstring

* Pr comments

---------
Co-authored-by: Ian Schweer <ischweer@riotgames.com>
Co-authored-by: Ricardo Vieira <28983449+ricardoV94@users.noreply.github.com>

This can only be done when the output of infer_shape of the core_op depends only on the input shapes, and not their values.

This Op does not really fit the CAReduce API, as it requires an extra bit of information (number of elements in the axis) during the loop. A better solution will be a fused Elemwise+CAReduce

* Remove opinionated message about ignore_duplicates.

Setting to false can lead to slower code on C/Numba backend which don't support np.add.at natively.

* Setting to false can lead to slower code on C/Numba backend which don't support np.add.at natively.

Support multidimensional boolean set/inc_subtensor in Numba via rewrite

Bumps [mamba-org/setup-micromamba](https://github.com/mamba-org/setup-micromamba) from 1 to 2.
- [Release notes](https://github.com/mamba-org/setup-micromamba/releases)
- [Commits](https://github.com/mamba-org/setup-micromamba/compare/v1...v2)

---
updated-dependencies:
- dependency-name: mamba-org/setup-micromamba
  dependency-type: direct:production
  update-type: version-update:semver-major
...
Signed-off-by: dependabot[bot] <support@github.com>

Bumps [actions/setup-python](https://github.com/actions/setup-python) from 4 to 5.
- [Release notes](https://github.com/actions/setup-python/releases)
- [Commits](https://github.com/actions/setup-python/compare/v4...v5)

---
updated-dependencies:
- dependency-name: actions/setup-python
  dependency-type: direct:production
  update-type: version-update:semver-major
...
Signed-off-by: dependabot[bot] <support@github.com>

Bumps [codecov/codecov-action](https://github.com/codecov/codecov-action) from 4 to 5.
- [Release notes](https://github.com/codecov/codecov-action/releases)
- [Changelog](https://github.com/codecov/codecov-action/blob/main/CHANGELOG.md)
- [Commits](https://github.com/codecov/codecov-action/compare/v4...v5)

---
updated-dependencies:
- dependency-name: codecov/codecov-action
  dependency-type: direct:production
  update-type: version-update:semver-major
...
Signed-off-by: dependabot[bot] <support@github.com>

Bumps [pypa/cibuildwheel](https://github.com/pypa/cibuildwheel) from 2.21.3 to 2.22.0.
- [Release notes](https://github.com/pypa/cibuildwheel/releases)
- [Changelog](https://github.com/pypa/cibuildwheel/blob/main/docs/changelog.md)
- [Commits](https://github.com/pypa/cibuildwheel/compare/v2.21.3...v2.22.0)

---
updated-dependencies:
- dependency-name: pypa/cibuildwheel
  dependency-type: direct:production
  update-type: version-update:semver-minor
...
Signed-off-by: dependabot[bot] <support@github.com>

* Allow for scipy module resolution

* Add softplus

* Add tests

* Allow scipy scalar handling

* Check for scipy in elemwise

---------
Co-authored-by: Ian Schweer <ischweer@riotgames.com>

Co-authored-by: Mohit Kumar <mohitk@apporchid.com>