A new PR to generalize the vmap behavior

flatironinstitute / jax-finufft

JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library

Apache License 2.0

77 stars 2 forks source link

Closed dfm closed 2 years ago

dfm commented 2 years ago

ref #7