xla
[torch_xla2] Implement `bessel_`, `chebyshev_`, `hermite_*`, `erfcx`, `i0e`
#7723

Merged

[torch_xla2] Implement `bessel_`, `chebyshev_`, `hermite_*`, `erfcx`, `i0e` #7723

qihqi merged 15 commits into master from wcromar/special-bessel

adapt bessel_j0 formula to JAX

bb41e4ee

use fori_loop

5fd942c2

remove `jnp.vectorize`

4a0041ac

bessel_j1

39d7bebb

bessel_y0 and bessel_y1

0067199f

Move `foreach_loop` to `op_base`

18945b96

Fix usage of piecewise

870d9723

use `promote_int_input`

039d1b44

will-cromar force pushed from eac6bf2c to 039d1b44 1 year ago

special.chebyshev_polynomial_t

b351de39

reference

26fee3ae

special_chebyshev_polynomial_u

008ad4dd

`special_hermite_polynomial_h`

0271296b

i0e

480c92eb

erfcx

8532ab96

will-cromar requested a review from

qihqi 1 year ago

will-cromar marked this pull request as ready for review 1 year ago

remove obsolete import

2951ded1

will-cromar changed the title [torch_xla2] Implement special.bessel_* [torch_xla2] Implement `bessel_*`, `chebyshev_*`, `hermite_*`, `erfcx`, `i0e` 1 year ago

qihqi approved these changes on 2024-07-26

qihqi merged 07a83636 into master 1 year ago

qihqi deleted the wcromar/special-bessel branch 1 year ago

ManfeiBai added torchxla2-op

miladm assigned

will-cromar 1 year ago

Reviewers

qihqi

Assignees

will-cromar

Labels

torchxla2-op

Milestone

No milestone

xla [torch_xla2] Implement `bessel_*`, `chebyshev_*`, `hermite_*`, `erfcx`, `i0e` #7723 Merged

[torch_xla2] Implement `bessel_*`, `chebyshev_*`, `hermite_*`, `erfcx`, `i0e` #7723

xla
[torch_xla2] Implement `bessel_`, `chebyshev_`, `hermite_*`, `erfcx`, `i0e`
#7723

Merged

[torch_xla2] Implement `bessel_`, `chebyshev_`, `hermite_*`, `erfcx`, `i0e` #7723