JAX filters

[alicjapolanska]

Finish JAX implementation of filters

[alicjapolanska]

Can't figure out why the installation is failing here, I didn't change anything from main in that aspect @CosmoMatt

[codecov-commenter]

Codecov Report

Patch coverage: 71.42% and project coverage change: -16.81 :warning:

Comparison is base (5fb38af) 94.62% compared to head (9c33a89) 77.81%.

:mega: This organization is not using Codecov’s GitHub App Integration. We recommend you install it so Codecov can continue to function properly for your repositories. Learn more

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #60 +/- ## =========================================== - Coverage 94.62% 77.81% -16.81% =========================================== Files 7 9 +2 Lines 279 550 +271 =========================================== + Hits 264 428 +164 - Misses 15 122 +107 ``` | [Impacted Files](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics) | Coverage Δ | | |---|---|---| | [s2wav/transforms/numpy\_scattering.py](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics#diff-czJ3YXYvdHJhbnNmb3Jtcy9udW1weV9zY2F0dGVyaW5nLnB5) | `18.60% <18.60%> (ø)` | | | [s2wav/transforms/jax\_scattering.py](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics#diff-czJ3YXYvdHJhbnNmb3Jtcy9qYXhfc2NhdHRlcmluZy5weQ==) | `20.00% <20.00%> (ø)` | | | [s2wav/transforms/jax\_wavelets.py](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics#diff-czJ3YXYvdHJhbnNmb3Jtcy9qYXhfd2F2ZWxldHMucHk=) | `79.41% <79.41%> (ø)` | | | [s2wav/filter\_factory/kernels.py](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics#diff-czJ3YXYvZmlsdGVyX2ZhY3Rvcnkva2VybmVscy5weQ==) | `90.56% <80.76%> (-9.44%)` | :arrow_down: | | [s2wav/utils/shapes.py](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics#diff-czJ3YXYvdXRpbHMvc2hhcGVzLnB5) | `91.30% <80.95%> (-8.70%)` | :arrow_down: | | [s2wav/filter\_factory/tiling.py](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics#diff-czJ3YXYvZmlsdGVyX2ZhY3RvcnkvdGlsaW5nLnB5) | `74.46% <85.71%> (+7.80%)` | :arrow_up: | | [s2wav/filter\_factory/filters.py](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics#diff-czJ3YXYvZmlsdGVyX2ZhY3RvcnkvZmlsdGVycy5weQ==) | `93.47% <96.29%> (+1.17%)` | :arrow_up: | | [s2wav/transforms/numpy\_wavelets.py](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics#diff-czJ3YXYvdHJhbnNmb3Jtcy9udW1weV93YXZlbGV0cy5weQ==) | `97.87% <97.87%> (ø)` | | | [s2wav/utils/math\_functions.py](https://codecov.io/gh/astro-informatics/s2wav/pull/60?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics#diff-czJ3YXYvdXRpbHMvbWF0aF9mdW5jdGlvbnMucHk=) | `100.00% <100.00%> (ø)` | | Help us with your feedback. Take ten seconds to tell us [how you rate us](https://about.codecov.io/nps?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics). Have a feature suggestion? [Share it here.](https://app.codecov.io/gh/feedback/?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=astro-informatics)

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Do you have feedback about the report comment? Let us know in this issue.